import gradio as gr
from gradio.mix import Parallel

examples = [['আমার সোনার বাংলা'],['মনে পড়ে, রুবি রায়']]

translator = gr.Interface.load("huggingface/Helsinki-NLP/opus-mt-bn-en",title="Translation")


io1 = gr.Interface.load("huggingface/flax-community/gpt2-bengali",
                 title="Bengali-GPT2")


io2 = gr.Interface.load("huggingface/khalidsaifullaah/bengali-lyricist-gpt2",
                       title = "Finetuned Bengali-GPT2(Song Lyrics)")



iface = Parallel(translator,io1,io2,
         title = "Bengali-gpt2 demo", 
         examples=examples,
         layout='vertical',
         description = "Features pretrained gpt2 bengali model along with finetuned version on song lyrics")

if __name__ == "__main__":
    iface.launch()