Hugging Face's logo

Spaces:
romnatall
/
team_nlp_project
Sleeping

App Files Community
romnatall commited on
Commit
e949db1
·
1 Parent(s): 26290c2

deploy

Browse files
Files changed (2) hide show
  1. pages/anti_toxic/anti_toxic.py +1 -1
  2. pages/anti_toxic/dont_be_toxic.pt +3 -0
pages/anti_toxic/anti_toxic.py CHANGED
@@ -7,7 +7,7 @@ import streamlit as st
7
  @st.cache_resource
8
  def load_model():
9
  catboost_model = CatBoostClassifier(random_seed=42,eval_metric='Accuracy')
10
- catboost_model.load_model("pages/anti_toxic/dont_be_toxic.cbm")
11
  model_checkpoint = 'cointegrated/rubert-tiny-toxicity'
12
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
13
  model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint)
 
7
  @st.cache_resource
8
  def load_model():
9
  catboost_model = CatBoostClassifier(random_seed=42,eval_metric='Accuracy')
10
+ catboost_model.load_model("pages/anti_toxic/dont_be_toxic.pt")
11
  model_checkpoint = 'cointegrated/rubert-tiny-toxicity'
12
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
13
  model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint)
pages/anti_toxic/dont_be_toxic.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9584499d739622440d6579e33375a391c66b033ff502e740146da1ad7a55ee92
3
+ size 1105984