TextSummarizer / app.py
yaminichandun's picture
Create app.py
aa1a2e4 verified
raw
history blame contribute delete
630 Bytes
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()