Update app.py

app.py

@@ -5,7 +5,6 @@ from diffusers import AutoencoderKL, TCDScheduler
 from diffusers.models.model_loading_utils import load_state_dict
 from gradio_imageslider import ImageSlider
 from huggingface_hub import hf_hub_download
-from transformers import pipeline
 
 from controlnet_union import ControlNetModel_Union
 from pipeline_fill_sd_xl import StableDiffusionXLFillPipeline
@@ -14,8 +13,13 @@ MODELS = {
     "RealVisXL V5.0 Lightning": "SG161222/RealVisXL_V5.0_Lightning",
 }
 
-# 번역 모델 로드
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+# Replace the problematic translation model with a simpler function
+def translate_if_korean(text):
+    # Just log that Korean was detected but return the original text
+    if any('\u3131' <= char <= '\u318E' or '\uAC00' <= char <= '\uD7A3' for char in text):
+        print(f"Korean text detected: {text}")
+        print("Translation is disabled - using original text")
+    return text
 
 config_file = hf_hub_download(
     "xinsir/controlnet-union-sdxl-1.0",
@@ -48,48 +52,44 @@ pipe = StableDiffusionXLFillPipeline.from_pretrained(
 
 pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
 
-def translate_if_korean(text):
-    # 입력된 텍스트가 한글을 포함하고 있는지 확인
-    if any('\u3131' <= char <= '\u318E' or '\uAC00' <= char <= '\uD7A3' for char in text):
-        # 한글이 포함되어 있다면 번역
-        translated = translator(text)[0]['translation_text']
-        print(f"Translated prompt: {translated}")  # 디버깅을 위한 출력
-        return translated
-    return text
-
 @spaces.GPU
 def fill_image(prompt, image, model_selection):
-    # 프롬프트 번역
+    # Translate prompt if needed
     translated_prompt = translate_if_korean(prompt)
     
-    (
-        prompt_embeds,
-        negative_prompt_embeds,
-        pooled_prompt_embeds,
-        negative_pooled_prompt_embeds,
-    ) = pipe.encode_prompt(translated_prompt, "cuda", True)
-
-    source = image["background"]
-    mask = image["layers"][0]
-
-    alpha_channel = mask.split()[3]
-    binary_mask = alpha_channel.point(lambda p: p > 0 and 255)
-    cnet_image = source.copy()
-    cnet_image.paste(0, (0, 0), binary_mask)
-
-    for image in pipe(
-        prompt_embeds=prompt_embeds,
-        negative_prompt_embeds=negative_prompt_embeds,
-        pooled_prompt_embeds=pooled_prompt_embeds,
-        negative_pooled_prompt_embeds=negative_pooled_prompt_embeds,
-        image=cnet_image,
-    ):
-        yield image, cnet_image
-
-    image = image.convert("RGBA")
-    cnet_image.paste(image, (0, 0), binary_mask)
-
-    yield source, cnet_image
+    try:
+        (
+            prompt_embeds,
+            negative_prompt_embeds,
+            pooled_prompt_embeds,
+            negative_pooled_prompt_embeds,
+        ) = pipe.encode_prompt(translated_prompt, "cuda", True)
+
+        source = image["background"]
+        mask = image["layers"][0]
+
+        alpha_channel = mask.split()[3]
+        binary_mask = alpha_channel.point(lambda p: p > 0 and 255)
+        cnet_image = source.copy()
+        cnet_image.paste(0, (0, 0), binary_mask)
+
+        for image in pipe(
+            prompt_embeds=prompt_embeds,
+            negative_prompt_embeds=negative_prompt_embeds,
+            pooled_prompt_embeds=pooled_prompt_embeds,
+            negative_pooled_prompt_embeds=negative_pooled_prompt_embeds,
+            image=cnet_image,
+        ):
+            yield image, cnet_image
+
+        image = image.convert("RGBA")
+        cnet_image.paste(image, (0, 0), binary_mask)
+
+        yield source, cnet_image
+    except Exception as e:
+        print(f"Error during image generation: {e}")
+        # Return the original image in case of error
+        return source, source
 
 def clear_result():
     return gr.update(value=None)
@@ -109,36 +109,36 @@ with gr.Blocks(theme="Yntec/HaleyCH_Theme_Orange", css=css) as demo:
     with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(
-                label="프롬프트",
-                info="마스크에 채워넣을 내용을 설명하세요 (한글 또는 영어)",
+                label="Prompt",
+                info="Describe what to fill in the mask area (Korean or English)",
                 lines=3,
             )
         with gr.Column():
             model_selection = gr.Dropdown(
                 choices=list(MODELS.keys()),
                 value="RealVisXL V5.0 Lightning",
-                label="모델",
+                label="Model",
             )
-            run_button = gr.Button("생성")
+            run_button = gr.Button("Generate")
 
     with gr.Row():
         input_image = gr.ImageMask(
             type="pil",
-            label="입력 이미지",
+            label="Input Image",
             crop_size=(1024, 1024),
             layers=False
         )
 
         result = ImageSlider(
             interactive=False,
-            label="생성된 이미지",
+            label="Generated Image",
         )
 
-    use_as_input_button = gr.Button("입력 이미지로 사용", visible=False)  
+    use_as_input_button = gr.Button("Use as Input Image", visible=False)  
 
-    # 샘플 이미지 추가
+    # Add sample image
     with gr.Row(elem_classes="sample-image"):
-        sample_image = gr.Image("sample.png", label="샘플 이미지", height=256, width=256)
+        sample_image = gr.Image("sample.png", label="Sample Image", height=256, width=256)
     
     def use_output_as_input(output_image):
         return gr.update(value=output_image[1])