Vadim Borisov
commited on
Commit
•
dd05fd7
1
Parent(s):
895effa
Update README.md
Browse files
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 |
|