Update app.py

app.py

@@ -29,8 +29,7 @@ def show_image_pair(pair_id):
     
     img1_path = os.path.join(current_dir, record['image1'])
     img2_path = os.path.join(current_dir, record['image2'])
-    img1 = gr.Image(label="Image 1", elem_id="img1", width=300)
-    img2 = gr.Image(label="Image 2", elem_id="img2", width=300)
+
 
     
     # Check if images exist
@@ -69,6 +68,8 @@ with gr.Blocks() as demo:
         with gr.Row():
             img1 = gr.Image(label="Image 1")
             img2 = gr.Image(label="Image 2")
+            img1 = gr.Image(label="Image 1", elem_id="img1", width=400)
+            img2 = gr.Image(label="Image 2", elem_id="img2", width=400)
     
     with gr.Row():
         prompt = gr.Textbox(label="Prompt", interactive=False)