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