Update app.py

app.py

@@ -22,15 +22,15 @@ 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)[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"
+        probs = torch.softmax(outputs.logits, dim=1)[0]
+        pred = torch.argmax(probs).item()
+        confidence = min(probs[pred].item(), 0.95)  # 限制置信度最大为95%
+        percent = round(confidence * 100, 2)
+        result = f"🧠 预测 / Prediction:\n{labels[pred]}: {percent}%"
         return result
 
 
+
 # Gradio 界面
 interface = gr.Interface(
     fn=classify,