TRACES commited on
Commit
7d555a4
·
1 Parent(s): 82947de

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -14,13 +14,13 @@ def load_models():
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_disinformation = pickle.load(f)
16
 
17
- with open('models/tfidf_vectorizer_untrue_inform_detection_tfidf_bg_0.96_F1_score.pkl', 'rb') as f:
18
  st.session_state.tfidf_vectorizer_untrue_inf = pickle.load(f)
19
 
20
  with open('models/svm_model_2_classes_gpt_chatgpt_detection_tfidf_bg_0.886_F1_score.pkl', 'rb') as f:
21
  st.session_state.gpt_detector = pickle.load(f)
22
 
23
- with open('models/SVM_model_untrue_inform_detection_tfidf_bg_0.96_F1_score.pkl', 'rb') as f:
24
  st.session_state.untrue_detector = pickle.load(f)
25
 
26
  st.session_state.bert = pipeline(task="text-classification",
 
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_disinformation = pickle.load(f)
16
 
17
+ with open('models/tfidf_vectorizer_untrue_inform_detection_tfidf_bg_0.96_F1_score_3Y_N_Q1_082023.pkl', 'rb') as f:
18
  st.session_state.tfidf_vectorizer_untrue_inf = pickle.load(f)
19
 
20
  with open('models/svm_model_2_classes_gpt_chatgpt_detection_tfidf_bg_0.886_F1_score.pkl', 'rb') as f:
21
  st.session_state.gpt_detector = pickle.load(f)
22
 
23
+ with open('models/SVM_model_untrue_inform_detection_tfidf_bg_0.96_F1_score_3Y_N_Q1_082023.pkl', 'rb') as f:
24
  st.session_state.untrue_detector = pickle.load(f)
25
 
26
  st.session_state.bert = pipeline(task="text-classification",