Gpt2-bengali / gradioapp.py
Tahsin-Mayeesha's picture
added streamlit and gradio app
2d78164
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()