elo4 commited on
Commit
0caed5c
·
verified ·
1 Parent(s): 66e8482

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -19,8 +19,8 @@ This is a fine-tuned TinyBERT model for sentiment analysis on the Tripadvisor da
19
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
20
 
21
  # Load the model
22
- tokenizer = AutoTokenizer.from_pretrained("username/tinybert-sentiment-analysis")
23
- model = AutoModelForSequenceClassification.from_pretrained("username/tinybert-sentiment-analysis")
24
 
25
  # Predict sentiment
26
  text = "The hotel was amazing and had great service!"
 
19
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
20
 
21
  # Load the model
22
+ tokenizer = AutoTokenizer.from_pretrained("elo4/TinyBERT-sentiment-model")
23
+ model = AutoModelForSequenceClassification.from_pretrained("elo4/TinyBERT-sentiment-model")
24
 
25
  # Predict sentiment
26
  text = "The hotel was amazing and had great service!"