File size: 637 Bytes
f5a1953
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
import torch 
import gradio as gr 

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']

gr.close_all()
app = gr.Interface(fn=summarize,inputs=gr.Textbox(label="Plug in your text here to summarize", lines=10, placeholder="Enter text to summarize..."), outputs=gr.Textbox(label="Summary", lines=5, placeholder="Summary will appear here..."), title="Text-Summarizer", description="This application summarizes the input text.", theme="default")
app.launch()