update
Browse files
README.md
CHANGED
|
@@ -33,5 +33,5 @@ predicted_class = logits.argmax(dim=-1).item()
|
|
| 33 |
|
| 34 |
sentiment = "Negative" if predicted_class == 0 else "Positive"
|
| 35 |
|
| 36 |
-
print(f"Predicted class: {predicted_class}
|
| 37 |
```
|
|
|
|
| 33 |
|
| 34 |
sentiment = "Negative" if predicted_class == 0 else "Positive"
|
| 35 |
|
| 36 |
+
print(f"Predicted class: {predicted_class}\nSentimen: {sentiment}")
|
| 37 |
```
|