Spaces:
Running
Running
Add ChatGPT detection
Browse files
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/
|
15 |
st.session_state.tfidf_vectorizer = pickle.load(f)
|
16 |
|
17 |
-
with open('models/
|
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_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",
|