fix

app.py

@@ -288,20 +288,6 @@ with gr.Blocks() as demo:
             #     outputs=[image_display_with_grid_aft, selected_cells_aft]
             # )
             
-            image_bef.change(
-                fn=None,
-                inputs=[image_bef],
-                outputs=[],
-                js="(image) => { initializeEditor(); importBackground(image); return []; }",
-            )
-
-            image_aft.change(
-                fn=None,
-                inputs=[image_aft],
-                outputs=[],
-                js="(image) => { initializeEditor(); importBackground(image); return []; }",
-            )
-            
         with gr.Column(scale=1):
             # Output components
             prediction = gr.Textbox(label="Predicted caption")
@@ -317,6 +303,20 @@ with gr.Blocks() as demo:
         outputs=[prediction, selected_info_bef, selected_info_aft]
     )
 
+    image_bef.change(
+                fn=None,
+                inputs=[image_bef],
+                outputs=[],
+                js="(image) => { initializeEditor(); importBackground(image); return []; }",
+            )
+
+    image_aft.change(
+        fn=None,
+        inputs=[image_aft],
+        outputs=[],
+        js="(image) => { initializeEditor(); importBackground(image); return []; }",
+    )
+
 
 
 app.mount("/js", StaticFiles(directory="js"), name="js")