debugging predictions.

app.py

@@ -114,7 +114,7 @@ def infer(video_file):
         logits = outputs.logits
     predicted_label = logits.argmax(-1).item()
     print(f"Predicted class: {MODEL.config.id2label[predicted_label]}")
-    softmax_scores = torch.nn.functional.softmax(logits, dim=0).squeeze(0)
+    softmax_scores = torch.nn.functional.softmax(logits, dim=-1).squeeze(0)
     print(f"Softmax scores: {softmax_scores}.")
     confidences = {LABELS[i]: float(softmax_scores[i]) for i in range(len(LABELS))}
     print(f"Confidence dict: {confidences}")