fix gr input issue

app.py

@@ -274,7 +274,10 @@ with gr.Blocks(title="TEASER demo", css="""
                 gr.Markdown("---")
                 gr.Markdown("### Example Source Images (Click to use)")
                 with gr.Row(elem_classes=["scrollable-column"]):
-                    example_images = [f for f in sorted(os.listdir(IN_IMAGE_DIR)) if f.endswith(('.png', '.jpg', '.jpeg'))]
+                    if os.path.exists(IN_IMAGE_DIR):
+                        example_images = [f for f in sorted(os.listdir(IN_IMAGE_DIR)) if f.endswith(('.png', '.jpg', '.jpeg'))]
+                    else:
+                        example_images = []
                     for img in example_images:
                         img_path = os.path.join(IN_IMAGE_DIR, img)
                         with gr.Column(elem_classes=["image-container"]):
@@ -288,7 +291,10 @@ with gr.Blocks(title="TEASER demo", css="""
                 gr.Markdown("---")
                 gr.Markdown("### Example Driven Images (Click to use)")
                 with gr.Row(elem_classes=["scrollable-column"]):
-                    example_images = [f for f in sorted(os.listdir(IN_IMAGE_DIR)) if f.endswith(('.png', '.jpg', '.jpeg'))]
+                    if os.path.exists(IN_IMAGE_DIR):
+                        example_images = [f for f in sorted(os.listdir(IN_IMAGE_DIR)) if f.endswith(('.png', '.jpg', '.jpeg'))]
+                    else:
+                        example_images = []
                     for img in example_images:
                         img_path = os.path.join(IN_IMAGE_DIR, img)
                         with gr.Column(elem_classes=["image-container"]):