Ifeanyi commited on
Commit
24324bc
·
1 Parent(s): cdfe23d

Update app.py

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