user commited on
Commit ·
b45da75
1
Parent(s): 28d574b
Class name
Browse files
README.md
CHANGED
|
@@ -150,7 +150,10 @@ with torch.no_grad():
|
|
| 150 |
logits = model(**inputs).logits
|
| 151 |
|
| 152 |
predicted_class_id = logits.argmax().item()
|
|
|
|
|
|
|
| 153 |
print(f"Predicted Class ID: {predicted_class_id}")
|
|
|
|
| 154 |
```
|
| 155 |
|
| 156 |
## Intended Uses & Limitations
|
|
|
|
| 150 |
logits = model(**inputs).logits
|
| 151 |
|
| 152 |
predicted_class_id = logits.argmax().item()
|
| 153 |
+
predicted_class_name = model.config.id2label[predicted_class_id]
|
| 154 |
+
|
| 155 |
print(f"Predicted Class ID: {predicted_class_id}")
|
| 156 |
+
print(f"Predicted Class Name: {predicted_class_name}")
|
| 157 |
```
|
| 158 |
|
| 159 |
## Intended Uses & Limitations
|