Update app.py

app.py

@@ -4,10 +4,17 @@ from PIL import Image
 import numpy as np
 import gradio as gr
 from trans import translate
-import urllib.request
 import os 
+from transformers import pipeline
 
 
+def translate(text):
+    model_checkpoint = "Helsinki-NLP/opus-mt-ko-en"
+    translator = pipeline("translation", model=model_checkpoint)
+    translated=translator(text)
+    translated=translated[0]["translation_text"]
+    return translated
+
 model_id = "stabilityai/stable-diffusion-xl-base-1.0"
 adapter_id = "ksyint/teu_lora"