sayakpaul HF staff commited on
Commit
33da5d3
1 Parent(s): e7f0154

debugging predictions.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -113,7 +113,7 @@ def infer(video_file):
113
  outputs = MODEL(**inputs)
114
  logits = outputs.logits
115
  predicted_label = logits.argmax(-1).item()
116
- print("Predicted class: {MODEL.config.id2label[predicted_label]}")
117
  softmax_scores = torch.nn.functional.softmax(logits, dim=0).squeeze(0)
118
  print(f"Softmax scores: {softmax_scores}.")
119
  confidences = {LABELS[i]: float(softmax_scores[i]) for i in range(len(LABELS))}
 
113
  outputs = MODEL(**inputs)
114
  logits = outputs.logits
115
  predicted_label = logits.argmax(-1).item()
116
+ print(f"Predicted class: {MODEL.config.id2label[predicted_label]}")
117
  softmax_scores = torch.nn.functional.softmax(logits, dim=0).squeeze(0)
118
  print(f"Softmax scores: {softmax_scores}.")
119
  confidences = {LABELS[i]: float(softmax_scores[i]) for i in range(len(LABELS))}