Update README.md
Browse files
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("
|
23 |
-
model = AutoModelForSequenceClassification.from_pretrained("
|
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!"
|