Update app.py

app.py

@@ -192,6 +192,7 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                 seed_output = gr.Textbox(label="Used Seed")
                 intermediate_output = gr.Gallery(label='Output', elem_id="gallery", visible=args.dev)
 
+
         with gr.Row():
             with gr.Column():
                 gr.Markdown("## Examples")
@@ -216,6 +217,7 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
 
                 # Gallery 컴포넌트를 사용하여 그리드 형태로 예시들을 표시합니다
                 gallery = gr.Gallery(
+                    value=list(zip(example_images, example_captions)),
                     label="Example Gallery",
                     show_label=False,
                     elem_id="gallery",
@@ -223,9 +225,7 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
                     rows=2,
                     object_fit="contain",
                     height="auto"
-                ).style(grid=5, height="auto")
-                
-                gallery.value = list(zip(example_images, example_captions))
+                )
 
                 # 예시 클릭 시 입력 필드에 값을 채우는 함수
                 def fill_example(evt: gr.SelectData):