Spaces:
Build error
Build error
| import gradio as gr | |
| from transformers import pipeline | |
| import torch | |
| summarizer = pipeline( | |
| "summarization", | |
| 'pszemraj/led-large-book-summary' | |
| ) | |
| def summarizer(text): | |
| result = summarizer(text) | |
| return result[0]["summary_text"] | |
| with gr.Blocks() as demo: | |
| gr.Markdown("#Text-to-Image Generator") | |
| text = gr.Textbox(label="Text Input", placeholder="Enter your text......") | |
| submit_btn = gr.Button("Summarize") | |
| summary = gr.Textbox(label="Summary") | |
| submit_btn.click(fn=summarizer, inputs=text, outputs=summary) | |
| demo.launch() |