Update app.py

app.py

@@ -34,9 +34,18 @@ class GlobalVars:
 
 g = GlobalVars()
 
+
+        
 def initialize_models(device):
     try:
         print("Initializing models...")
+        g.translator = transformers_pipeline(
+            "translation", 
+            model="Helsinki-NLP/opus-mt-ko-en",
+            device=device
+        )
+        print("Model initialization completed successfully")
+        
         # 3D 생성 파이프라인
         g.trellis_pipeline = TrellisImageTo3DPipeline.from_pretrained(
             "JeffreyXiang/TRELLIS-image-large"
@@ -368,8 +377,10 @@ def text_to_image(prompt: str, height: int, width: int, steps: int, scales: floa
             return any(ord('가') <= ord(c) <= ord('힣') for c in text)
         
         if contains_korean(prompt):
+            # Helsinki-NLP/opus-mt-ko-en 모델을 사용하여 번역
             translated = g.translator(prompt)[0]['translation_text']
             prompt = translated
+            print(f"Translated prompt: {prompt}")
         
         formatted_prompt = f"wbgmsst, 3D, {prompt}, white background"