Update app.py

app.py

@@ -1120,7 +1120,22 @@ if __name__ == "__main__":
             examples_instance.load_from_cache = MethodType(custom_load_from_cache, examples_instance)
             
         apply_custom_load_from_cache(examples)
-
+        
+        # Connect the event handler for file upload changes
+        gallery.change(
+            fn=update_gallery_state,
+            inputs=[gallery, images_state],
+            outputs=[images_state, gallery, preview_gallery],
+            show_progress="full"
+        ).then(
+            fn=update_height_width,
+            inputs=[images_state],
+            outputs=[height, width]
+        ).then(
+            fn=update_denoise_checkboxes,
+            inputs=[images_state, task_type, azimuth, elevation, distance],
+            outputs=[denoise_mask_checkbox]
+        )
       
 
     demo.launch()