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()