TEST_translate / app.py
GiladtheFixer's picture
Update app.py
e59973f
raw
history blame
407 Bytes
import gradio as gr
from transformers import pipeline
def translate(text,number):
if number==1:
model='Helsinki-NLP/opus-mt-en-es'
else if number==2:
model='Helsinki-NLP/opus-mt-en-fr'
else:
model='Helsinki-NLP/opus-mt-en-ru'
pipe=pipeline(model)
return pipe(text)
demo = gr.Interface(fn=translate, inputs="text,number", outputs="json")
demo.launch()