Update app.py

app.py

@@ -163,6 +163,7 @@ def infer_canny(prompt, text_for_image, text_position, font_size,
           control_guidance_end = 0.9,
           strength = 1.0
         ):
+
     prompt = translate_korean_to_english(prompt)
     negative_prompt = translate_korean_to_english(negative_prompt)
 
@@ -176,6 +177,7 @@ def infer_canny(prompt, text_for_image, text_position, font_size,
 
     pipe = pipe_canny.to("cuda")
     condi_img = process_canny_condition(init_image)
+ 
     image = pipe(
         prompt=prompt,
         image=init_image,
@@ -189,9 +191,7 @@ def infer_canny(prompt, text_for_image, text_position, font_size,
         num_images_per_prompt=1,
         generator=generator,
     ).images[0]
-    return [condi_img, image], seed
-
-
+    return image, seed  # CANNY 이미지 반환 제거
 
 def update_button_states(selected_position):
     return [
@@ -329,8 +329,9 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as Kolors:
             with gr.Row():
                 canny_button = gr.Button("Start", elem_id="button")
 
+
         with gr.Column(elem_id="col-right"):
-            result = gr.Gallery(label="Result", show_label=False, columns=2)
+            result = gr.Image(label="Result", show_label=False)  # Gallery에서 Image로 변경
             seed_used = gr.Number(label="Seed Used")
 
     canny_button.click(
@@ -338,6 +339,8 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as Kolors:
         inputs = [prompt, text_for_image, text_position, font_size, negative_prompt, seed, randomize_seed, guidance_scale, num_inference_steps, controlnet_conditioning_scale, control_guidance_end, strength],
         outputs = [result, seed_used]
     )
+        
+
 
     # Set initial button states
     Kolors.load(update_button_states, inputs=[text_position], outputs=position_buttons)