import gradio as gr
from transformers import pipeline
from model import *

model = build_transformer(2000, 2500, 200, 200)
translator = pipeline('translation_en_to_fr')

def translate(text):
    result = translator(text)[0]['translation_text']
    return result

examples = [
    "Hi, How are you?",
    "I don't understand.",
    "I am sorry."
]

demo = gr.Interface(
    fn = translate, 
    inputs = [
        gr.Textbox(label = "Enter Sentence")
    ],
    outputs = [
        gr.Textbox(label="Translated Text")
    ],
    examples = examples,
    cache_examples = True
)

demo.launch()