import gradio as gr
from transformers import pipeline


model_checkpoint = "huynguyen208/fantastic4-finetuned-en-to-vi-demo"
translator = pipeline("translation", model=model_checkpoint)

def translate(sentence):
    return translator(sentence)

iface = gr.Interface(fn=translate,
                     inputs="text",
                     outputs="text",
                     title = 'Translate en to vi',
                     description = 'Mini Translator')
iface.launch(inline = False)