Update app.py

app.py

@@ -23,12 +23,22 @@ def classify(text):
     with torch.no_grad():
         outputs = model(**inputs)
         probs = torch.softmax(outputs.logits, dim=1)[0]
-        pred = torch.argmax(probs).item()
-        confidence = min(probs[pred].item(), 0.95)  # 限制置信度最大为95%
+        
+        threshold = 0.7  # 自定义阈值（你可以改成别的）
+        if probs[1].item() > threshold:
+            pred = 1  # 判为操纵性
+        else:
+            pred = 0  # 判为非操纵性
+
+        confidence = min(probs[pred].item(), 0.95)  # 置信度依然可以控制上限
+
         percent = round(confidence * 100, 2)
         result = f"Prediction / 预测:\n{labels[pred]}\n📊 Confidence /置信度: {percent}%"
         return result
 
+#谁大选谁        
+#pred = torch.argmax(probs).item()
+#confidence = min(probs[pred].item(), 0.95)  # 限制置信度最大为95%
 
 # Gradio 界面
 interface = gr.Interface(