import gradio as gr
from gradio import inputs
from gradio.mix import Parallel

description = 'Comparing dialogue summarization models'

io1 = gr.Interface.load("huggingface/anegi/t5smallmodel",
                        description = 'This is a self trained model',
                        title = 'Dialogue Summarization using t5-small model',
                        inputs = [
                            gr.inputs.Textbox(lines = 7, label = 'Text', placeholder = 'Enter your text here: ')  
                        ],
                        theme = 'dark-peach'
                        )
                        
io2 = gr.Interface.load("huggingface/philschmid/bart-large-cnn-samsum",
                        description = 'This is a pre trained model',
                        title = 'Dialogue Summarization using bart large cnn',
                        inputs = [
                            gr.inputs.Textbox(lines = 7, label = 'Text', placeholder = 'Enter your text here: ')  
                        ],
                        theme = 'dark-peach'
                        )
                        
Parallel(io1,io2).launch()