Adapt the box sizes.

app.py

@@ -65,8 +65,8 @@ def pred_maps(image_1, image_2, image_3, image_4):
 N = 4
 # examples = [[_] for _ in glob('example_images/butterfly/*')][:N]
 
-ipt = [gr.Image(width=600, height=150) for _ in range(N)]
-opt = [gr.Image(width=600, height=150) for _ in range(N)]
+ipt = [gr.Image(width=600, height=300) for _ in range(N)]
+opt = [gr.Image(width=600, height=300) for _ in range(N)]
 demo = gr.Interface(
     fn=pred_maps,
     inputs=ipt,