Spaces:
Sleeping
Sleeping
reduce processing time
Browse files
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=
|
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 |
)
|