Aryanikale23 commited on
Commit
5a7272e
1 Parent(s): 670cf35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -108,7 +108,7 @@ def infer(video_file):
108
  logits = outputs.logits
109
  softmax_scores = torch.nn.functional.softmax(logits, dim=-1).squeeze(0)
110
  confidences = {LABELS[i]: float(softmax_scores[i]) for i in range(len(LABELS))}
111
- return {confidences}
112
 
113
  gr.Interface(
114
  fn=infer,
 
108
  logits = outputs.logits
109
  softmax_scores = torch.nn.functional.softmax(logits, dim=-1).squeeze(0)
110
  confidences = {LABELS[i]: float(softmax_scores[i]) for i in range(len(LABELS))}
111
+ return confidences
112
 
113
  gr.Interface(
114
  fn=infer,