Update main.py

main.py

@@ -55,14 +55,13 @@ if all([
     # st.session_state.gpt_detector_result = ''
     # st.session_state.gpt_detector_probability = [1, 0]
 
-    st.session_state.bert_gpt_result = [{'label': '', 'score': 1}]
 
     st.session_state.untrue_detector_result = ''
     st.session_state.untrue_detector_probability = 1
     
     st.session_state.bert_disinfo_result = [{'label': '', 'score': 1}]
         
-
+    st.session_state.bert_gpt_result = [{'label': '', 'score': 1}]
 
 content = load_content()
 if 'loaded' not in st.session_state:
@@ -103,8 +102,7 @@ if st.session_state.agree:
             # st.session_state.gpt_detector_result = st.session_state.gpt_detector.predict(user_tfidf_disinformation)[0]
             # st.session_state.gpt_detector_probability = st.session_state.gpt_detector.predict_proba(user_tfidf_disinformation)[0]
 
-            st.session_state.bert_gpt_result = st.session_state.bert_gpt(user_input)
-
+            
             user_tfidf_untrue_inf = st.session_state.tfidf_vectorizer_untrue_inf.transform([user_input])
             st.session_state.untrue_detector_result = st.session_state.untrue_detector.predict(user_tfidf_untrue_inf)[0]
             st.session_state.untrue_detector_probability = st.session_state.untrue_detector.predict_proba(user_tfidf_untrue_inf)[0]
@@ -112,6 +110,8 @@ if st.session_state.agree:
 
             st.session_state.bert_disinfo_result = st.session_state.bert_disinfo(user_input)
 
+            st.session_state.bert_gpt_result = st.session_state.bert_gpt(user_input)
+
             
     
         # if st.session_state.gpt_detector_result == 1:
@@ -123,15 +123,7 @@ if st.session_state.agree:
         #                str(round(st.session_state.gpt_detector_probability[0] * 100, 2)) +
         #                content['gpt_no_proba'][st.session_state.lang], icon="✅")
 
-        if st.session_state.bert_gpt_result[0]['label'] == 'LABEL_1':
-            st.warning(content['bert_gpt_1'][st.session_state.lang] +
-                       str(round(st.session_state.bert_gpt_result[0]['score'] * 100, 2)) +
-                       content['bert_gpt_2'][st.session_state.lang], icon = "⚠️")
-        else:
-            st.success(content['bert_human_1'][st.session_state.lang] +
-                       str(round(st.session_state.bert_gpt_result[0]['score'] * 100, 2)) +
-                       content['bert_human_2'][st.session_state.lang], icon="✅")
-
+        
         if st.session_state.untrue_detector_result == 0:
             st.warning(content['untrue_getect_yes'][st.session_state.lang] +
                        str(round(st.session_state.untrue_detector_probability * 100, 2)) +
@@ -150,6 +142,16 @@ if st.session_state.agree:
                        str(round(st.session_state.bert_disinfo_result[0]['score'] * 100, 2)) +
                        content['bert_no_2'][st.session_state.lang], icon="✅")
 
+        if st.session_state.bert_gpt_result[0]['label'] == 'LABEL_1':
+            st.warning(content['bert_gpt_1'][st.session_state.lang] +
+                       str(round(st.session_state.bert_gpt_result[0]['score'] * 100, 2)) +
+                       content['bert_gpt_2'][st.session_state.lang], icon = "⚠️")
+        else:
+            st.success(content['bert_no_1'][st.session_state.lang] +
+                       str(round(st.session_state.bert_gpt_result[0]['score'] * 100, 2)) +
+                       content['bert_no_2'][st.session_state.lang], icon="✅")
+
+
     
         st.info(content['disinformation_definition'][st.session_state.lang], icon="ℹ️")