TRACES commited on
Commit
a2f9b23
1 Parent(s): bc34d7b

Add new models

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -11,10 +11,10 @@ from sklearn.feature_extraction.text import TfidfVectorizer
11
  def load_models():
12
  st.session_state.loaded = True
13
 
14
- with open("models/tfidf_vectorizer_svm_model_gpt_chatgpt_detection_tfidf_bg_0.88F1_score.pkl", "rb") as f:
15
  st.session_state.tfidf_vectorizer = pickle.load(f)
16
 
17
- with open('models/svm_model_gpt_chatgpt_detection_tfidf_bg_0.88F1_score.pkl', 'rb') as f:
18
  st.session_state.gpt_detector = pickle.load(f)
19
 
20
  st.session_state.bert = pipeline(task="text-classification",
 
11
  def load_models():
12
  st.session_state.loaded = True
13
 
14
+ with open("models/tfidf_vectorizer_svm_model_2_classes_gpt_chatgpt_detection_tfidf_bg_0.886_F1_score.pkl", "rb") as f:
15
  st.session_state.tfidf_vectorizer = pickle.load(f)
16
 
17
+ with open('models/svm_model_2_classes_gpt_chatgpt_detection_tfidf_bg_0.886_F1_score.pkl', 'rb') as f:
18
  st.session_state.gpt_detector = pickle.load(f)
19
 
20
  st.session_state.bert = pipeline(task="text-classification",