Update app.py

app.py

@@ -14,6 +14,10 @@ import random
 import time
 import requests
 import pandas as pd
+from transformers import pipeline
+
+# 번역 모델 로드
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 
 #Load prompts for randomization
 df = pd.read_csv('prompts.csv', header=None)
@@ -309,12 +313,18 @@ def generate_image_to_image(prompt_mash, image_input_path, image_strength, steps
     ).images[0]
     return final_image
 
+
 def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_indices, lora_scale_1, lora_scale_2, randomize_seed, seed, width, height, loras_state, progress=gr.Progress(track_tqdm=True)):
+    # 한글 감지 및 번역
+    if any('\u3131' <= char <= '\u318E' or '\uAC00' <= char <= '\uD7A3' for char in prompt):
+        translated = translator(prompt, max_length=512)[0]['translation_text']
+        print(f"Original prompt: {prompt}")
+        print(f"Translated prompt: {translated}")
+        prompt = translated
+
     if not selected_indices:
         raise gr.Error("You must select at least one LoRA before proceeding.")
 
-    selected_loras = [loras_state[idx] for idx in selected_indices]
-
     # Build the prompt with trigger words
     prepends = []
     appends = []