unnir commited on
Commit
dd05fd7
1 Parent(s): 895effa

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -68,7 +68,7 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name)
68
 
69
  # Function to predict sentiment
70
  def predict_sentiment(text):
71
- inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True, max_length=512)
72
  with torch.no_grad():
73
  outputs = model(**inputs)
74
 
 
68
 
69
  # Function to predict sentiment
70
  def predict_sentiment(text):
71
+ inputs = tokenizer(text.lower(), return_tensors="pt", truncation=True, padding=True, max_length=512)
72
  with torch.no_grad():
73
  outputs = model(**inputs)
74