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 demo: | |
textbox=gr.Textbox(placeholder="Enter text block to summarize",lines=4) | |
gr.Interface(fn=predict,inputs=textbox,outputs="text") | |
demo.launch() |