Spaces:
Runtime error
Runtime error
debugging predictions.
Browse files
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))}
|