Update app.py

app.py

@@ -76,7 +76,7 @@ def classify_email_single_model(text, model_name):
     with torch.no_grad():
         outputs = models[model_name](**inputs)
         prediction = torch.argmax(outputs.logits, dim=1).item()
-    probs = F.softmax(logits, dim=1)
+    probs = F.softmax(outputs.logits, dim=1)
     probs_percent = probs.cpu().numpy() * 100
     response = {"prediction": labels[prediction], "probabilities": probs_percent}
     return response