Update app.py

app.py

@@ -131,7 +131,7 @@ def predict(tweets):
             probs = torch.softmax(outs[0], dim=-1)
             preds = torch.argmax(probs, dim=-1)
             for i in range(len(tweets)):
-                res.append({'TWEETS': tweets, 'LABEL': preds[i].item(), 'PROBABILITY': probs[i][preds[i]].item()})
+                res.append({'TWEETS': tweets, 'LABEL': preds[i].item(), 'PROBABILITY': probs[i][preds[i].item()].item()})
     return res
 
 res = predict(testing_loader)