AriusXi's picture
Update app.py
a80789b
from transformers import pipeline
import gradio as grad
mdl_name = "Helsinki-NLP/opus-mt-ar-en"
opus_translator = pipeline("translation", model=mdl_name)
def translate(text):
response = opus_translator(text)
return response
grad.Interface(translate, inputs=["text",], outputs="text").launch()