import gradio as gr
from Text_Summarization import summarize

gr.close_all()
demo = gr.Interface(
    fn=summarize, 
    inputs=[gr.Textbox(label='Text to Summarize', lines=6)], 
    outputs=[gr.Textbox(label='Result', lines=3)],
    title='Text Summarization with distilBART-cnn',
    description="Summarize any given text using `sshleifer/distilbart-cnn-12-6` model under the hood!"
    )
demo.launch()