sohan-ai commited on
Commit
8618136
1 Parent(s): b5234b6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 = "username/repo_name"
25
- tokenizer = DistilBertTokenizer.from_pretrained(model_name)
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