Update app.py
Browse files
app.py
CHANGED
|
@@ -46,7 +46,7 @@ def predict(sentence, entity, model_choice):
|
|
| 46 |
entity,
|
| 47 |
return_tensors="pt",
|
| 48 |
truncation=True,
|
| 49 |
-
max_length=
|
| 50 |
)
|
| 51 |
|
| 52 |
with torch.inference_mode():
|
|
|
|
| 46 |
entity,
|
| 47 |
return_tensors="pt",
|
| 48 |
truncation=True,
|
| 49 |
+
max_length=160
|
| 50 |
)
|
| 51 |
|
| 52 |
with torch.inference_mode():
|