File size: 805 Bytes
2d78164
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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()