Spaces:

romnatall
/

team_nlp_project

Sleeping

romnatall commited on Apr 11

Commit

e949db1

•

1 Parent(s): 26290c2

deploy

Files changed (2) hide show

pages/anti_toxic/anti_toxic.py CHANGED Viewed

@@ -7,7 +7,7 @@ import streamlit as st
 @st.cache_resource
 def load_model():
     catboost_model = CatBoostClassifier(random_seed=42,eval_metric='Accuracy')
-    catboost_model.load_model("pages/anti_toxic/dont_be_toxic.cbm")
     model_checkpoint = 'cointegrated/rubert-tiny-toxicity'
     tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
     model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint)

 @st.cache_resource
 def load_model():
     catboost_model = CatBoostClassifier(random_seed=42,eval_metric='Accuracy')
+    catboost_model.load_model("pages/anti_toxic/dont_be_toxic.pt")
     model_checkpoint = 'cointegrated/rubert-tiny-toxicity'
     tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
     model = AutoModelForSequenceClassification.from_pretrained(model_checkpoint)

pages/anti_toxic/dont_be_toxic.pt ADDED Viewed

+version https://git-lfs.github.com/spec/v1
+oid sha256:9584499d739622440d6579e33375a391c66b033ff502e740146da1ad7a55ee92
+size 1105984