Spaces:
Running
Running
kovacsvi
commited on
Commit
·
ed88b8b
1
Parent(s):
10307a1
use padding
Browse files- interfaces/cap.py +1 -1
interfaces/cap.py
CHANGED
@@ -105,7 +105,7 @@ def predict(text, model_id, tokenizer_id):
|
|
105 |
text,
|
106 |
max_length=256,
|
107 |
truncation=True,
|
108 |
-
padding=
|
109 |
return_tensors="pt"
|
110 |
)
|
111 |
inputs = {k: v.to(device) for k, v in inputs.items()}
|
|
|
105 |
text,
|
106 |
max_length=256,
|
107 |
truncation=True,
|
108 |
+
padding=True,
|
109 |
return_tensors="pt"
|
110 |
)
|
111 |
inputs = {k: v.to(device) for k, v in inputs.items()}
|