Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ async def predict(request: TextRequest):
|
|
31 |
inputs = tokenizer.encode_plus(
|
32 |
request.text,
|
33 |
add_special_tokens=True,
|
34 |
-
max_length=
|
35 |
truncation=True,
|
36 |
padding='max_length',
|
37 |
return_attention_mask=True,
|
@@ -54,7 +54,7 @@ async def batch_predict(request: BatchTextRequest):
|
|
54 |
inputs = tokenizer.encode_plus(
|
55 |
text,
|
56 |
add_special_tokens=True,
|
57 |
-
max_length=
|
58 |
truncation=True,
|
59 |
padding='max_length',
|
60 |
return_attention_mask=True,
|
|
|
31 |
inputs = tokenizer.encode_plus(
|
32 |
request.text,
|
33 |
add_special_tokens=True,
|
34 |
+
max_length=64,
|
35 |
truncation=True,
|
36 |
padding='max_length',
|
37 |
return_attention_mask=True,
|
|
|
54 |
inputs = tokenizer.encode_plus(
|
55 |
text,
|
56 |
add_special_tokens=True,
|
57 |
+
max_length=64,
|
58 |
truncation=True,
|
59 |
padding='max_length',
|
60 |
return_attention_mask=True,
|