Add fine-tuned EuroBERT for binary geopolitical classification
Browse files
README.md
CHANGED
|
@@ -47,7 +47,7 @@ texts = [
|
|
| 47 |
|
| 48 |
]
|
| 49 |
|
| 50 |
-
inputs = tokenizer(texts, padding=True, truncation=True, max_length=
|
| 51 |
|
| 52 |
with torch.no_grad():
|
| 53 |
logits = model(**inputs).logits
|
|
|
|
| 47 |
|
| 48 |
]
|
| 49 |
|
| 50 |
+
inputs = tokenizer(texts, padding=True, truncation=True, max_length=512, return_tensors="pt")
|
| 51 |
|
| 52 |
with torch.no_grad():
|
| 53 |
logits = model(**inputs).logits
|