import torch
import gradio as gr

from transformers import pipeline

text_summary = pipeline(task="summarization", model="sshleifer/distilbart-cnn-12-6",
                        dtype=torch.bfloat16)
def Summary(input):
    output=text_summary(input)
    return output[0]["summary_text"]

gr.close_all()

demo = gr.Interface(fn=Summary,
                    inputs=[gr.Textbox(label="Input text to summarize", lines=10)],
                    outputs=[gr.Textbox(label="Summarized text", lines=4)],
                    title="Sravan GENAI Project 1: Text Summarization",
                    description="This app will be used to summarize the text")
demo.launch()