TEST_translate / app.py
GiladtheFixer's picture
Update app.py
5f75b38
raw
history blame
777 Bytes
# 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()
import gradio as gr
from transformers import pipeline
pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-en-ru")
def predict(text):
return pipe(text)[0]["translation_text"]
demo = gr.Interface(
fn=predict,
inputs='text',
outputs='text',
)
demo.launch()