Spaces:
Sleeping
Sleeping
vkovacs commited on
Commit ·
e3bfea3
1
Parent(s): dd02147
64 tokens might be enough for sentences
Browse files
app.py
CHANGED
|
@@ -47,7 +47,7 @@ def predict(text, model_id, tokenizer_id):
|
|
| 47 |
model.to(device)
|
| 48 |
|
| 49 |
inputs = tokenizer(text,
|
| 50 |
-
max_length=
|
| 51 |
truncation=True,
|
| 52 |
padding="do_not_pad",
|
| 53 |
return_tensors="pt").to(device)
|
|
|
|
| 47 |
model.to(device)
|
| 48 |
|
| 49 |
inputs = tokenizer(text,
|
| 50 |
+
max_length=64,
|
| 51 |
truncation=True,
|
| 52 |
padding="do_not_pad",
|
| 53 |
return_tensors="pt").to(device)
|