Spaces:

King-8
/

help-classifier-api

Sleeping

King-8 commited on 4 days ago

Commit

a471ca8

verified ·

1 Parent(s): 2e88ae2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -34,8 +34,9 @@ def predict(input: InputText):
     probs = F.softmax(outputs.logits, dim=1)
     predicted_class = probs.argmax().item()
     return {
-        "label": labels[predicted_class]
     }

     probs = F.softmax(outputs.logits, dim=1)
     predicted_class = probs.argmax().item()
+    confidence = probs.max().item()
     return {
+        "label": labels[predicted_class],
+        "confidence": round(confidence, 4)
     }