File size: 630 Bytes
aa1a2e4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import torch
import gradio

from transformers import pipeline

text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)

def summarize(text):
    output=text_summary(text)
    return output[0]['summary_text']

gradio.close_all()
demo=gradio.Interface(fn=summarize, inputs=[gradio.Textbox(label="Input your text to summarize",lines=6)],
                      outputs=[gradio.Textbox(label="Summarized Text",lines=4)],
                      title="Text Summarization Application",
                      description="This application is used to summarize your text.")

demo.launch()