meta-artem commited on
Commit
01f683c
1 Parent(s): a61b6f7

Add len elements

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -2,13 +2,16 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
 
5
- def main(article):
6
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
7
 
8
- return summarizer(article, max_length=130, min_length=30, do_sample=False)[0]['summary_text']
9
 
10
 
11
  iface = gr.Interface(fn=main,
12
- inputs=gr.inputs.Textbox(lines=5, label="Article text"),
 
 
 
13
  outputs="text")
14
  iface.launch()
2
  from transformers import pipeline
3
 
4
 
5
+ def main(article, min_len, max_len):
6
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
7
 
8
+ return summarizer(article, max_length=max_len, min_length=min_len, do_sample=False)[0]["summary_text"]
9
 
10
 
11
  iface = gr.Interface(fn=main,
12
+ inputs=[gr.inputs.Textbox(lines=5, label="Article text"),
13
+ "number",
14
+ "number"
15
+ ],
16
  outputs="text")
17
  iface.launch()