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()