Update README.md
Browse files
README.md
CHANGED
@@ -49,4 +49,5 @@ with torch.no_grad():
|
|
49 |
|
50 |
predicted_label = model.config.id2label[torch.argmax(outputs.logits).item()]
|
51 |
print(predicted_label)
|
|
|
52 |
```
|
|
|
49 |
|
50 |
predicted_label = model.config.id2label[torch.argmax(outputs.logits).item()]
|
51 |
print(predicted_label)
|
52 |
+
# --> high
|
53 |
```
|