Jaehan's picture
Update app.py
7f2a4c3
raw
history blame contribute delete
No virus
308 Bytes
from transformers import pipeline
import gradio as grad
model_name = "Helsinki-NLP/opus-mt-ko-en"
opus_translator = pipeline("translation", model=model_name)
def translate(text):
response = opus_translator(text)
return response
grad.Interface(translate, inputs=["text"], outputs=["text"]).launch()