Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ def predict(image):
|
|
| 27 |
logits = outputs.logits
|
| 28 |
probs = torch.nn.functional.softmax(logits,dim=1).squeeze()
|
| 29 |
top_idx = probs.argmax().item()
|
| 30 |
-
label = model.config.
|
| 31 |
confidence = probs[top_idx].item()
|
| 32 |
return f"{label} ({round(confidence*100, 1)}%)"
|
| 33 |
|
|
|
|
| 27 |
logits = outputs.logits
|
| 28 |
probs = torch.nn.functional.softmax(logits,dim=1).squeeze()
|
| 29 |
top_idx = probs.argmax().item()
|
| 30 |
+
label = model.config.id2label[top_idx]
|
| 31 |
confidence = probs[top_idx].item()
|
| 32 |
return f"{label} ({round(confidence*100, 1)}%)"
|
| 33 |
|