Update pages/文件式文本检测工具.py

pages/文件式文本检测工具.py

@@ -78,9 +78,9 @@ class Model(nn.Module):
     
 def load_models_and_predict(text, device):
     # 加载模型
-    MacBERT_base_CDialBias = torch.load('hjp157688\ROCAI\models\MacBERT-base-CDialBias.pth')
+    MacBERT_base_CDialBias = torch.load('ROCAI\models\MacBERT-base-CDialBias.pth')
     MacBERT_base_CDialBias.to(device)
-    MacBERT_base_COLD = torch.load('hjp157688\ROCAI\models\MacBERT-base-CDialBias.pth')
+    MacBERT_base_COLD = torch.load('ROCAI\models\MacBERT-base-CDialBias.pth')
     MacBERT_base_COLD.to(device)
 
     # 获取密钥和ID
@@ -181,7 +181,10 @@ if file is not None:
     # 添加保存结果的选项
     save_results = st.checkbox("保存结果为CSV文件")
     
-    if st.button("开始检测") and st.session_state.logged_in ==  True:
+    if st.button("开始检测") :
+        if st.session_state.logged_in = False:
+            st.error("请先登录！")
+            st.stop()
         if column not in df.columns:
             st.error(f"列名 '{column}' 不存在于数据集中，请检查并重新输入。")
         else:
@@ -241,7 +244,4 @@ if file is not None:
                     file_name='results.csv',
                     mime='text/csv'
                 )
-    elif st.button("开始检测") and st.session_state.logged_in == False:
-        st.error("请先登录！")
-        st.stop()