Update README.md
Browse files
README.md
CHANGED
@@ -82,7 +82,7 @@ features = tokenizer(*tokenized_pairs, padding=True, truncation="longest_first",
|
|
82 |
|
83 |
model.eval()
|
84 |
with torch.no_grad():
|
85 |
-
model_predictions =
|
86 |
logits = activation_fct(model_predictions.logits)
|
87 |
|
88 |
scores = [score[0] for score in logits]
|
|
|
82 |
|
83 |
model.eval()
|
84 |
with torch.no_grad():
|
85 |
+
model_predictions = model(**features, return_dict=True)
|
86 |
logits = activation_fct(model_predictions.logits)
|
87 |
|
88 |
scores = [score[0] for score in logits]
|