Update app.py

app.py

@@ -7,10 +7,6 @@ import time
 import os
 from diffusers import DiffusionPipeline
 from custom_pipeline import FLUXPipelineWithIntermediateOutputs
-from transformers import pipeline
-
-# 번역 모델 설정 (CPU 사용)
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device="cpu")
 
 # 상수 정의
 MAX_SEED = np.iinfo(np.int32).max
@@ -20,6 +16,11 @@ DEFAULT_HEIGHT = 1024
 DEFAULT_INFERENCE_STEPS = 1
 GPU_DURATION = 15  # GPU 할당 시간 축소
 
+# 간단한 한글 감지 함수
+def is_korean(text):
+    """한글 포함 여부 확인"""
+    return any('\u3131' <= char <= '\u3163' or '\uac00' <= char <= '\ud7a3' for char in text)
+
 # 모델 설정
 def setup_model():
     dtype = torch.float16
@@ -45,16 +46,6 @@ labels = {
     "Inspiration Gallery": "영감 갤러리"
 }
 
-def translate_if_korean(text):
-    """한글 텍스트를 영어로 안전하게 번역"""
-    try:
-        if any('\u3131' <= char <= '\u3163' or '\uac00' <= char <= '\ud7a3' for char in text):
-            return translator(text)[0]['translation_text']
-        return text
-    except Exception as e:
-        print(f"번역 오류: {e}")
-        return text
-
 # 이미지 생성 함수
 @spaces.GPU(duration=GPU_DURATION)
 def generate_image(prompt, seed=None, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT, 
@@ -64,8 +55,10 @@ def generate_image(prompt, seed=None, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT
         if not isinstance(seed, (int, type(None))):
             seed = None
             randomize_seed = True
-            
-        prompt = translate_if_korean(prompt)
+        
+        # 한글 프롬프트에 대한 경고 메시지 추가
+        if is_korean(prompt):
+            print("경고: 한글 프롬프트는 직접 처리됩니다. 번역기를 사용하지 않습니다.")
         
         if seed is None or randomize_seed:
             seed = random.randint(0, MAX_SEED)
@@ -179,11 +172,6 @@ def create_snow_effect():
     
     return gr.HTML(snow_html)
 
-# Gradio 앱에서 사용할 때:
-# with app:  아래에
-
-
-    
 # Gradio UI 구성
 with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
     
@@ -283,5 +271,4 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
-
+    demo.launch()