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 |
+
|