Update app.py
Browse files
app.py
CHANGED
|
@@ -60,7 +60,7 @@ def predict(sentence, entity, model_choice):
|
|
| 60 |
entity,
|
| 61 |
return_tensors="pt",
|
| 62 |
truncation=True,
|
| 63 |
-
max_length=
|
| 64 |
)
|
| 65 |
|
| 66 |
with torch.inference_mode():
|
|
|
|
| 60 |
entity,
|
| 61 |
return_tensors="pt",
|
| 62 |
truncation=True,
|
| 63 |
+
max_length=160
|
| 64 |
)
|
| 65 |
|
| 66 |
with torch.inference_mode():
|