Update app.py

app.py

@@ -278,19 +278,30 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 lines=3
             )
             
-            # Examples 갤러리 추가
+            # Examples 갤러리를 text_prompt 아래로 이동
             if example_images:
-                gr.Gallery(
-                    value=example_images,
-                    label="Example Images",
-                    show_label=True,
-                    elem_id="gallery",
-                    columns=[3],
-                    rows=[2],
-                    height=200,
-                    allow_preview=True
-                )
-                
+                with gr.Row():
+                    gallery = gr.Gallery(
+                        value=example_images,
+                        label="Example Images",
+                        show_label=True,
+                        elem_id="gallery",
+                        columns=[3],
+                        rows=[2],
+                        height=200,
+                        allow_preview=True
+                    ).style(grid=3)
+                    
+                    # Gallery 클릭 이벤트 추가
+                    def load_example(evt: gr.SelectData):
+                        return example_images[evt.index]
+                    
+                    gallery.select(
+                        load_example,
+                        None,
+                        image_prompt,
+                        show_progress=True
+                    )
             
             with gr.Accordion("Image Generation Settings", open=False):
                 with gr.Row():
@@ -343,16 +354,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 slat_sampling_steps = gr.Slider(1, 50, label="Latent Sampling Steps", value=12, step=1)
 
             generate_3d_btn = gr.Button("Generate 3D")
-
-            gr.Examples(
-                examples=example_images,
-                inputs=image_prompt,
-                outputs=[trial_id, image_prompt],
-                fn=preprocess_image,
-                cache_examples=True,
-                label="Click on an example to load it"
-            )
-
             
             with gr.Accordion("GLB Extraction Settings", open=False):
                 mesh_simplify = gr.Slider(0.9, 0.98, label="Simplify", value=0.95, step=0.01)
@@ -379,7 +380,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         outputs=[trial_id, image_prompt]
     )
 
-# 나머지 핸들러들
+    # 나머지 핸들러들
     image_prompt.upload(
         preprocess_image,
         inputs=[image_prompt],
@@ -431,6 +432,7 @@ if __name__ == "__main__":
         # CUDA 메모리 초기화
         if torch.cuda.is_available():
             torch.cuda.empty_cache()
+            torch.cuda.synchronize()
         
         # 초기 이미지 전처리 테스트
         try:
@@ -440,7 +442,12 @@ if __name__ == "__main__":
             print(f"Warning: Initial preprocessing test failed: {e}")
             
         # Gradio 인터페이스 실행
-        demo.launch(allowed_paths=[PERSISTENT_DIR])
+        demo.launch(
+            allowed_paths=[PERSISTENT_DIR],
+            enable_queue=True,
+            max_threads=4,
+            show_error=True
+        )
         
     except Exception as e:
         print(f"Error during initialization: {e}")