Spaces:
Runtime error
Runtime error
from transformers import pipeline | |
import gradio as gr | |
model = pipeline("summarization") | |
def predict(prompt: str) -> str: | |
summary = model(prompt)[0]["summary_text"] | |
return summary | |
# create an interface for the model | |
with gr.Interface(predict, "textbox", "text") as interface: | |
interface.launch() | |