File size: 561 Bytes
8507bad
 
9954c84
8507bad
 
 
 
 
 
245cf3e
75f1037
66ebae9
8507bad
 
 
 
 
 
245cf3e
8507bad
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import gradio as gr
from transformers import pipeline
import torch

summarizer = pipeline(
    "summarization",
    'pszemraj/led-large-book-summary'
)

def summarize_text(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=summarize_text, inputs=text, outputs=summary)
demo.launch()