from transformers import pipeline | |
import gradio as gr | |
model = pipeline("summarization") | |
def predict(prompt): | |
summary = model(prompt)[0]["summary_text"] | |
return summary | |
with gr.Blocks() as space: | |
textbox = gr.Textbox(placeholder="Enter text block to summarize", lines=4) | |
gr.Interface(fn=predict, inputs=textbox, outputs="text") | |
space.launch() |