Soumil30 commited on
Commit
9883bb6
·
1 Parent(s): 837d7a4

increased max words

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
5
 
6
  def summarize(input):
7
- return summarizer(input, max_length=100)[0]['summary_text']
8
 
9
  iface = gr.Interface(fn=summarize, inputs="text", outputs="text")
10
  iface.launch()
 
4
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
5
 
6
  def summarize(input):
7
+ return summarizer(input, max_length=250)[0]['summary_text']
8
 
9
  iface = gr.Interface(fn=summarize, inputs="text", outputs="text")
10
  iface.launch()