Spaces:
Runtime error
Runtime error
Commit
•
5a7272e
1
Parent(s):
670cf35
Update app.py
Browse files
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
|
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,
|