Update app.py

app.py

@@ -703,6 +703,8 @@ HEADER = """
 </style>
 """
 
+# ... [Previous imports and code unchanged until the Gradio Blocks interface] ...
+
 # Gradio web interface
 try:
     logger.info("Initializing Gradio Blocks interface")
@@ -736,7 +738,7 @@ try:
                                 minimum=8,
                                 maximum=50,
                                 step=1,
-                                value=30,  # Reduced for L4 and ZeroGPU
+                                value=30,
                                 info="Higher steps enhance detail but increase processing time",
                                 elem_classes="gr-slider"
                             )
@@ -766,11 +768,28 @@ try:
                         download_button = gr.Button("Download GLB", variant="secondary", elem_classes="gr-button-secondary")
             with gr.Tab("Cosmic Gallery"):
                 gr.Markdown("### Discover Stellar Creations")
+                # Ensure example directory exists and contains valid images
+                example_dir = f"{TRIPOSG_CODE_DIR}/assets/example_data"
+                examples = []
+                if os.path.exists(example_dir):
+                    valid_extensions = (".png", ".jpg", ".jpeg")
+                    examples = [
+                        [
+                            os.path.join(example_dir, image),  # image
+                            0,                                 # seed
+                            30,                                # num_inference_steps
+                            7.5,                               # guidance_scale
+                            True,                              # reduce_face
+                            DEFAULT_FACE_NUMBER,              # target_face_num
+                            "None"                            # style_filter
+                        ]
+                        for image in os.listdir(example_dir)
+                        if image.lower().endswith(valid_extensions)
+                    ]
+                if not examples:
+                    logger.warning(f"No valid images found in {example_dir}, skipping examples")
                 gr.Examples(
-                    examples=[
-                        f"{TRIPOSG_CODE_DIR}/assets/example_data/{image}"
-                        for image in os.listdir(f"{TRIPOSG_CODE_DIR}/assets/example_data")
-                    ],
+                    examples=examples,
                     fn=run_full,
                     inputs=[image_prompts, seed, num_inference_steps, guidance_scale, reduce_face, target_face_num, style_filter],
                     outputs=[seg_image, model_output, textured_model_output],