stestoni91 commited on
Commit
19bab30
1 Parent(s): bc5adae

Use airKlizz/mt5-base-wikinewssum-italian

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', model="google/pegasus-xsum")
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="airKlizz/mt5-base-wikinewssum-italian")
5
 
6
  def predict(text):
7
  summary = model(text)[0]['summary_text']