Update app.py

app.py

@@ -27,11 +27,11 @@ def classify_confidence(text):
         logits = outputs.logits
         probs = torch.nn.functional.softmax(logits, dim=1)
         predicted_class = torch.argmax(probs).item()
-        prediction = label_map[predicted_class]
         label = id2label[predicted_class]
+        prediction = label_map[label]
         score = round(probs[0][predicted_class].item() * 100, 2)
 
-    return f"{label} ({score}%)"
+    return f"{prediction} ({score}%)"
 
 iface = gr.Interface(
     fn=classify_confidence,