Spaces:

srjosueaaron
/

summarization-demo

Running

srjosueaaron commited on 16 days ago

Commit

ef1ab48

verified ·

1 Parent(s): 6dc0526

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 from transformers import pipeline
 import gradio as gr
-summarizer = pipeline("summarization", model="josmunpen/mt5-small-spanish-summarization", framework="pt")
 def generate_summary(text):
     summary = summarizer(text, max_length=100, min_length=30, do_sample=False)

 from transformers import pipeline
 import gradio as gr
+summarizer = pipeline("summarization", model="mrm8488/t5-small-spanish-summarization")
 def generate_summary(text):
     summary = summarizer(text, max_length=100, min_length=30, do_sample=False)