Einmalumdiewelt commited on
Commit
8cb95e9
·
1 Parent(s): 1439700

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def summarize(inputs):
15
  # with a max_length of 200, the model has a chance to encapsule lots of information
16
  preds = model.generate(**inputs,max_length=200,min_length=100)
17
  # we decode the predictions to store them
18
- decoded_predictions = tokenizer.batch_decode(predictions, skip_special_tokens=True)
19
  # each batches predictions are appended to the list
20
  return decoded_predictions
21
 
 
15
  # with a max_length of 200, the model has a chance to encapsule lots of information
16
  preds = model.generate(**inputs,max_length=200,min_length=100)
17
  # we decode the predictions to store them
18
+ decoded_predictions = tokenizer.batch_decode(preds, skip_special_tokens=True)
19
  # each batches predictions are appended to the list
20
  return decoded_predictions
21