Update app.py

app.py

@@ -195,19 +195,20 @@ with gr.Blocks(title="Animal Classifier", css=css) as demo:
             elem_classes=["centered-examples"]
         )
 
-    # Handle image selection from examples
+    # Handle image selection from examples - FIXED OUTPUTS
     def select_example(evt: gr.SelectData):
-        return example_images[evt.index]
+        selected_path = example_images[evt.index]
+        return selected_path, selected_path  # Return both image preview and state
     
     examples_gallery.select(
         fn=select_example,
-        outputs=[image_preview, current_image],
+        outputs=[image_preview, current_image],  # Match both components
         show_progress=False
     )
-
-    # Handle custom uploads
+    
+    # Fix upload handler too
     upload_btn.upload(
-        fn=lambda file: (file.name, file.name),
+        fn=lambda file: (file.name, file.name),  # Return both preview and state
         inputs=upload_btn,
         outputs=[image_preview, current_image]
     )