Spaces:
Running
Running
Update main.py
Browse files
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.
|
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.
|
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",
|