AyushDey commited on
Commit
fbcbcc2
·
1 Parent(s): 21d3046

Changed model to bart-large-cnn

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- pipe = pipeline('summarization')
5
 
6
  def summary(input):
7
  value = pipe(input)
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ pipe = pipeline('summarization', model='facebook/bart-large-cnn')
5
 
6
  def summary(input):
7
  value = pipe(input)