debugging predictions.

app.py

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