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

Fix len error

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import pipeline
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,
5
  def main(article, min_len, max_len):
6
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
7
 
8
+ return summarizer(article, max_length=int(max_len), min_length=int(min_len), do_sample=False)[0]["summary_text"]
9
 
10
 
11
  iface = gr.Interface(fn=main,