Update app.py

app.py

@@ -207,12 +207,8 @@ footer {
 }
 """
 
-
-
 with gr.Blocks(css=css) as demo:
     with gr.Column():
-
-
         with gr.Row():
             with gr.Column():
                 input_image = gr.Image(
@@ -248,14 +244,14 @@ with gr.Blocks(css=css) as demo:
                                 minimum=256,
                                 maximum=1536,
                                 step=8,
-                                value=720,  # Set a default value
+                                value=720,
                             )
                             height_slider = gr.Slider(
                                 label="Height",
                                 minimum=256,
                                 maximum=1536,
                                 step=8,
-                                value=480,  # Set a default value
+                                value=480,
                             )
                         with gr.Row():
                             num_inference_steps = gr.Slider(label="Steps", minimum=4, maximum=12, step=1, value=8)
@@ -275,15 +271,12 @@ with gr.Blocks(css=css) as demo:
 
                             custom_resize_size = gr.Slider(
                                 label="Custom resize size",
-                                minimum=256,  # 이전에는 720이었습니다.
+                                minimum=256,
                                 maximum=1024,
                                 step=8,
-                                value=512,  # 초기 값이므로 이 값도 적절히 조정할 수 있습니다.
-                                visible=False  # 기본적으로 이 슬라이더는 숨겨져 있으며, 'Custom'이 선택될 때만 보입니다.
+                                value=512,
+                                visible=False
                             )
-
-                            
-                            
                 gr.Examples(
                     examples=[
                         ["./examples/example_2.jpg", 1440, 810, "Middle"],
@@ -347,6 +340,7 @@ with gr.Blocks(css=css) as demo:
         inputs=[input_image, width_slider, height_slider, overlap_width, num_inference_steps,
                 resize_option, custom_resize_size, prompt_input, alignment_dropdown],
         outputs=result,
+        stream=True  # 추가: infer 함수가 제너레이터이므로 스트리밍 활성화
     ).then(
         fn=lambda: gr.update(visible=True),
         inputs=None,
@@ -362,11 +356,11 @@ with gr.Blocks(css=css) as demo:
         inputs=[input_image, width_slider, height_slider, overlap_width, num_inference_steps,
                 resize_option, custom_resize_size, prompt_input, alignment_dropdown],
         outputs=result,
+        stream=True  # 추가: 제너레이터 함수이므로 스트리밍 활성화
     ).then(
         fn=lambda: gr.update(visible=True),
         inputs=None,
         outputs=use_as_input_button,
     )
 
-
-demo.queue(max_size=12).launch(share=False)
+demo.queue(max_size=12).launch(share=False)