Spaces:
Sleeping
Sleeping
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() |