fix

app.py

@@ -249,29 +249,14 @@ with gr.Blocks() as demo:
             html_text = f"""
             <div id="container">
                 <canvas id="before" width="{width}" height="{height}"></canvas><img id="canvas-before" style="display:none;"/>
+                <br>
+                <br>
                 <canvas id="after" width="{width}" height="{height}"></canvas><img id="canvas-after" style="display:none;"/>
             </div>
             """
 
             html = gr.HTML(html_text)
 
-
-            # image_display_with_grid_bef = gr.Image(type="pil", label="Before Image with Grid")
-            # image_display_with_grid_aft = gr.Image(type="pil", label="After Image with Grid")
-            
-            # # Add click event to the displayed image
-            # image_display_with_grid_bef.select(
-            #     handle_click,
-            #     inputs=[image_display_with_grid_bef, selected_cells_bef, image_bef],
-            #     outputs=[image_display_with_grid_bef, selected_cells_bef]
-            # )
-
-            # image_display_with_grid_aft.select(
-            #     handle_click,
-            #     inputs=[image_display_with_grid_aft, selected_cells_aft, image_aft],
-            #     outputs=[image_display_with_grid_aft, selected_cells_aft]
-            # )
-
     with gr.Row():
         with gr.Column(scale=1):
             # Example images
@@ -285,18 +270,6 @@ with gr.Blocks() as demo:
                 examples_per_page=5
                 )
             
-            # image_bef.change(
-            #     fn=display_image,
-            #     inputs=[image_bef],
-            #     outputs=[image_display_with_grid_bef, selected_cells_bef]
-            # )
-
-            # image_aft.change(
-            #     fn=display_image,
-            #     inputs=[image_aft],
-            #     outputs=[image_display_with_grid_aft, selected_cells_aft]
-            # )
-            
         with gr.Column(scale=1):
             # Output components
             prediction = gr.Textbox(label="Predicted caption")