sayakpaul HF staff commited on
Commit
ffb234a
1 Parent(s): dfa0336

debugging confidence dict.

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -114,6 +114,7 @@ def infer(video_file):
114
  logits = outputs.logits
115
  softmax_scores = torch.nn.functional.softmax(logits, dim=0).squeeze(0)
116
  confidences = {LABELS[i]: float(softmax_scores[i]) for i in range(len(LABELS))}
 
117
  return confidences
118
 
119
 
 
114
  logits = outputs.logits
115
  softmax_scores = torch.nn.functional.softmax(logits, dim=0).squeeze(0)
116
  confidences = {LABELS[i]: float(softmax_scores[i]) for i in range(len(LABELS))}
117
+ print(f"Confidence dict: {confidences}")
118
  return confidences
119
 
120