Update README.md
Browse files
README.md
CHANGED
|
@@ -44,8 +44,7 @@ def predict_text(text: str):
|
|
| 44 |
text,
|
| 45 |
return_tensors="pt",
|
| 46 |
truncation=True,
|
| 47 |
-
padding="
|
| 48 |
-
max_length=512,
|
| 49 |
is_split_into_words=False
|
| 50 |
).to(device)
|
| 51 |
|
|
|
|
| 44 |
text,
|
| 45 |
return_tensors="pt",
|
| 46 |
truncation=True,
|
| 47 |
+
padding="longest",
|
|
|
|
| 48 |
is_split_into_words=False
|
| 49 |
).to(device)
|
| 50 |
|