Abijith commited on
Commit
809d949
1 Parent(s): ecff4a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ model = T5ForConditionalGeneration.from_pretrained(model_name)
14
  def summarize_text(input_text):
15
  summar_input = 'summarize: '+input_text
16
  input_tokens = tokenizer(summar_input, return_tensors='pt').input_ids
17
- outputs = model.generate(input_tokens)
18
  return tokenizer.decode(outputs[0], skip_special_tokens=True)
19
 
20
  # Interface for the Gradio app
 
14
  def summarize_text(input_text):
15
  summar_input = 'summarize: '+input_text
16
  input_tokens = tokenizer(summar_input, return_tensors='pt').input_ids
17
+ outputs = model.generate(input_tokens, max_new_tokens=100, min_new_tokens=30, do_sample=False)
18
  return tokenizer.decode(outputs[0], skip_special_tokens=True)
19
 
20
  # Interface for the Gradio app