nayeems94 commited on
Commit
f3d4702
·
verified ·
1 Parent(s): d5dab89

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
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/my-goemotions-finetuned")
31
- tokenizer = AutoTokenizer.from_pretrained("nayeems94/my-goemotions-finetuned")
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
+