timdpaep's picture
Changed language
8ab7dc3
from transformers import pipeline
import gradio as gr
language_translation_ckpt = f"Helsinki-NLP/opus-mt-nl-en"
translator = pipeline("translation", model=language_translation_ckpt)
def translate_nl_en(text):
translation = translator(text)
translation = translation[0]['translation_text']
return translation
with gr.Blocks() as demo:
nl_text = gr.Textbox(label="NL")
en_text = gr.Textbox(label="EN")
translate_button = gr.Button("Vertaal")
translate_button.click(translate_nl_en, inputs=nl_text, outputs=en_text, api_name=translate_nl_en)
demo.launch()