Update README.md
Browse files
README.md
CHANGED
@@ -21,8 +21,8 @@ Copy code
|
|
21 |
from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
|
22 |
|
23 |
# Load the fine-tuned model from Hugging Face
|
24 |
-
model_name = "
|
25 |
-
tokenizer = DistilBertTokenizer.from_pretrained(
|
26 |
model = DistilBertForSequenceClassification.from_pretrained(model_name)
|
27 |
|
28 |
# Tokenize input text
|
|
|
21 |
from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
|
22 |
|
23 |
# Load the fine-tuned model from Hugging Face
|
24 |
+
model_name = "sohan-ai/sentiment-analysis-model-amazon-reviews"
|
25 |
+
tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
|
26 |
model = DistilBertForSequenceClassification.from_pretrained(model_name)
|
27 |
|
28 |
# Tokenize input text
|