VietCat commited on
Commit
08a672b
·
1 Parent(s): 8a05f36

reduce processing time

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ async def summarize(req: Request, body: SummarizeRequest):
54
  # num_return_sequences=1
55
  # )
56
  outputs = model.generate(
57
- input_ids=input_ids, attention_mask=attention_masks,
58
  max_length=256,
59
  early_stopping=True
60
  )
 
54
  # num_return_sequences=1
55
  # )
56
  outputs = model.generate(
57
+ input_ids=input_ids, attention_mask=attention_mask,
58
  max_length=256,
59
  early_stopping=True
60
  )