Fix: model dependency + default sample

app.py

@@ -41,10 +41,14 @@ with gr.Blocks() as interface:
         sample_selection = gr.Dropdown(
             choices=list(sample_images.keys()),
             label="Select a Sample Image",
-            type="value",
+            type="Sample 1",
         )
         sample_display = gr.Image(label="Sample Image Preview", value=default_sample_image)
-        sample_selection.change(fn=load_sample_image, inputs=sample_selection, outputs=sample_display)
+        sample_selection.change(
+            fn=lambda sample_name: load_sample_image(sample_name),
+            inputs=sample_selection,
+            outputs=sample_display,
+        )
 
     selected_models = gr.CheckboxGroup(
         choices=["yolov5", "yolov8s"],
@@ -53,12 +57,17 @@ with gr.Blocks() as interface:
     )
 
     result_gallery = gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500)
-    print(uploaded_image)
-    print(sample_display)
-    gr.Button("Run").click(
-        fn=process_image,
-        inputs=[uploaded_image, selected_models],        
-        outputs=result_gallery,
-    )
+    if uploaded_image is not None:
+        gr.Button("Run").click(
+            fn=process_image,
+            inputs=[uploaded_image, selected_models],        
+            outputs=result_gallery,
+        )
+    else:
+        gr.Button("Run").click(
+            fn=process_image,
+            inputs=[sample_display, selected_models],        
+            outputs=result_gallery,
+        )
 
 interface.launch()