stestoni91 commited on
Commit
3218885
1 Parent(s): e88d8da

Use google/pegasus-xsum

Browse files
__pycache__/app.cpython-310.pyc ADDED
Binary file (669 Bytes). View file
 
__pycache__/test_app.cpython-310-pytest-8.2.2.pyc ADDED
Binary file (1.15 kB). View file
 
app.py CHANGED
@@ -1,7 +1,7 @@
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']
 
1
  from transformers import pipeline
2
  import gradio as gr
3
 
4
+ model = pipeline('summarization', model="google/pegasus-xsum")
5
 
6
  def predict(text):
7
  summary = model(text)[0]['summary_text']