raul-padua commited on
Commit
6cadcce
1 Parent(s): 43aec6d

Update app.py

Browse files

updating to base model.

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import pipeline
2
  import gradio as gr
3
 
4
- get_completion = pipeline("summarization", model="facebook/bart-large-cnn")
5
 
6
  def summarize(input):
7
  output = get_completion(input)
@@ -11,6 +11,6 @@ demo = gr.Interface(fn=summarize,
11
  inputs=[gr.Textbox(label="Text to summarize", lines=6)],
12
  outputs=[gr.Textbox(label="Result", lines=3)],
13
  title="Text summarization with distilbart-cnn",
14
- description="Summarize any text using the `facebook/bart-large-cnn` model, from Meta AI, under the hood!"
15
  )
16
  demo.launch(share=True)
 
1
  from transformers import pipeline
2
  import gradio as gr
3
 
4
+ get_completion = pipeline("summarization", model="facebook/bart-base-cnn")
5
 
6
  def summarize(input):
7
  output = get_completion(input)
 
11
  inputs=[gr.Textbox(label="Text to summarize", lines=6)],
12
  outputs=[gr.Textbox(label="Result", lines=3)],
13
  title="Text summarization with distilbart-cnn",
14
+ description="Summarize any text using the `facebook/bart-base-cnn` model, from Meta AI, under the hood!"
15
  )
16
  demo.launch(share=True)