Update app.py

app.py

@@ -21,11 +21,15 @@ labels = ["Non-manipulative / 非操纵性", "Manipulative / 操纵性"]
 def classify(text):
     inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True).to(device)
     with torch.no_grad():
-        outputs = model(**inputs)
-        probs = torch.softmax(outputs.logits, dim=1)
-        pred = torch.argmax(probs, dim=1).item()
-        confidence = probs[0][pred].item()
-        return f"🧠 预测 / Prediction: {labels[pred]}\n🔢 置信度 / Confidence: {confidence*100:.2f}%"
+    outputs = model(**inputs)
+    probs = torch.softmax(outputs.logits, dim=1)[0]  # 取第一个样本的概率向量
+    probs = torch.clamp(probs, max=0.95)  # 限制最大置信度为 95%
+    result = "🧠 预测 / Prediction:\n"
+    for i, label in enumerate(labels):
+        percent = round(probs[i].item() * 100, 2)
+        result += f"{label}: {percent}%\n"
+    return result
+
 
 # Gradio 界面
 interface = gr.Interface(