textsummarizer / app.py
cygon24's picture
Update app.py
557d7e6 verified
raw
history blame contribute delete
683 Bytes
import os
# Install dependencies manually if not found
os.system("pip install torch torchvision torchaudio transformers gradio")
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 summary(input):
output = text_summary(input)
return output[0]['summary_text']
demo = gr.Interface(
fn=summary,
inputs=[gr.Textbox(label="Input text to summarize", lines=7)],
outputs=[gr.Textbox(label="Summarized text", lines=4)],
title="@cygon: Text summarizer",
description="SIMPLE UNPAID APPLICATION FOR TEXT SUMMARIZING"
)
demo.launch()