Bijoy09 commited on
Commit
9639e29
1 Parent(s): 41034fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=128,
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=128,
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,