emilyyy04
/

burmese-sentiment-xlm-roberta

Text Classification

Model card Files Files and versions

emilyyy04 commited on Aug 10

Commit

f835215

·

verified ·

1 Parent(s): 5584103

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -114,4 +114,4 @@ outputs = model(**inputs)
 predicted_class = torch.argmax(outputs.logits, dim=1).item()
 label_map = {0: "positive", 1: "negative", 2: "neutral"}
-print("Predicted Sentiment:", label_map[predicted_class])

 predicted_class = torch.argmax(outputs.logits, dim=1).item()
 label_map = {0: "positive", 1: "negative", 2: "neutral"}
+print("Predicted Sentiment:", label_map[predicted_class])