Translation / app.py
VarunAIML's picture
code refactoring
e11824b verified
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()