Spaces:
Runtime error
Runtime error
File size: 470 Bytes
6e05fea e59973f 25b8940 e59973f 0126e4b 61ef2ea 0126e4b e59973f 0126e4b e59973f 0126e4b 6e05fea 1a0e861 6e05fea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import gradio as gr
from transformers import pipeline
def translate(text,number):
model1=''
if number==1:
model1='Helsinki-NLP/opus-mt-en-es'
elif number==2:
model1='Helsinki-NLP/opus-mt-en-fr'
else:
model1='Helsinki-NLP/opus-mt-en-ru'
pipe=pipeline("translation", model=model1)
return pipe(text)[0]["translation_text"]
demo = gr.Interface(fn=translate, inputs=["text","number"], outputs="json")
demo.launch() |