Update app.py

app.py

@@ -28,11 +28,11 @@ def select_image(image_path):
 with gr.Blocks() as iface:
     with gr.Row():
         with gr.Column():
-            selected_image = gr.Image(label="Selected Image")
+            selected_image = gr.Image(label="Selected Image", width=400, height=400)
             img_selector = gr.Dropdown(choices=img_files, label="Select an image from folder")
             img_selector.change(select_image, img_selector, selected_image)
         with gr.Column():
-            output_image = gr.Image(label="Output Image")
+            output_image = gr.Image(label="Output Image", width=800, height=800)
             predict_button = gr.Button("Predict")
             predict_button.click(predict_single_image, inputs=selected_image, outputs=output_image)