Update README.md
Browse files
README.md
CHANGED
@@ -19,7 +19,7 @@ sentences = [' what was the reason for everyone to leave the company ',
|
|
19 |
' what was the reasons for everyone to leave the company ',
|
20 |
' what were the reasons for everyone to leave the company ']
|
21 |
|
22 |
-
features = tokenizer(
|
23 |
model.eval()
|
24 |
with torch.no_grad():
|
25 |
scores = model(**features).logits
|
|
|
19 |
' what was the reasons for everyone to leave the company ',
|
20 |
' what were the reasons for everyone to leave the company ']
|
21 |
|
22 |
+
features = tokenizer(sentences, padding=True, truncation=True, return_tensors="pt")
|
23 |
model.eval()
|
24 |
with torch.no_grad():
|
25 |
scores = model(**features).logits
|