Ifeanyi commited on
Commit
06f7a94
·
1 Parent(s): 5d1445a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,12 +4,12 @@ import torch
4
 
5
  def textSummarizer(ARTICLE,OutputLength):
6
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
7
- result = summarizer(ARTICLE, max_length=244, min_length=OutputLength, do_sample=False)
8
  return result[0]["summary_text"]
9
 
10
 
11
  app = gr.Interface(textSummarizer,
12
- inputs = [gr.Textbox(),gr.Slider(minimum=0,maximum=50,step=1)],
13
  outputs = gr.Textbox(),
14
  title = "Article Summarizer",
15
  theme=gr.themes.Soft())
 
4
 
5
  def textSummarizer(ARTICLE,OutputLength):
6
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
7
+ result = summarizer(ARTICLE, max_length=OutputLength, min_length=50, do_sample=False)
8
  return result[0]["summary_text"]
9
 
10
 
11
  app = gr.Interface(textSummarizer,
12
+ inputs = [gr.Textbox(),gr.Slider(minimum=0,maximum=150,step=1)],
13
  outputs = gr.Textbox(),
14
  title = "Article Summarizer",
15
  theme=gr.themes.Soft())