Spaces:
Runtime error
Runtime error
import gradio as gr | |
from transformers import pipeline | |
#hub_model_id = "huggingface-course/mt5-small-finetuned-amazon-en-es" | |
#model_id = "philschmid/bart-large-cnn-samsum" | |
model_id = "t5-base" | |
summarizer = pipeline("summarization", model=model_id) | |
def summarize(text): | |
#return "Summary: " + text | |
return "Summary: " + str(summarizer(text)[0]["summary_text"]) | |
iface = gr.Interface(fn=summarize, inputs="text", outputs="text") | |
iface.launch() |