Spaces:

srjosueaaron
/

summarization-demo

Running

srjosueaaron commited on 18 days ago

Commit

6dc0526

verified ·

1 Parent(s): eba95f5

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")
 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="josmunpen/mt5-small-spanish-summarization", framework="pt")
 def generate_summary(text):
     summary = summarizer(text, max_length=100, min_length=30, do_sample=False)