Update README.md
Browse files
README.md
CHANGED
@@ -27,8 +27,8 @@ The model predicts the following emotions:
|
|
27 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
28 |
|
29 |
# Load model and tokenizer
|
30 |
-
model = AutoModelForSequenceClassification.from_pretrained("nayeems94/
|
31 |
-
tokenizer = AutoTokenizer.from_pretrained("nayeems94/
|
32 |
|
33 |
# Example text
|
34 |
text = "I am feeling so frustrated and angry!"
|
@@ -51,3 +51,4 @@ id2label = {
|
|
51 |
}
|
52 |
|
53 |
print(f"Predicted emotion: {id2label[predicted_class_id]}")
|
|
|
|
27 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
28 |
|
29 |
# Load model and tokenizer
|
30 |
+
model = AutoModelForSequenceClassification.from_pretrained("nayeems94/text-emotion-classifier")
|
31 |
+
tokenizer = AutoTokenizer.from_pretrained("nayeems94/text-emotion-classifier")
|
32 |
|
33 |
# Example text
|
34 |
text = "I am feeling so frustrated and angry!"
|
|
|
51 |
}
|
52 |
|
53 |
print(f"Predicted emotion: {id2label[predicted_class_id]}")
|
54 |
+
|