stestoni91 commited on
Commit
4738e5b
1 Parent(s): f2891ff

Use facebook/bart-large-cnn

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