Set 4 images as input.

app.py

@@ -34,8 +34,8 @@ if os.path.exists(state_dict):
 model.eval()
 
 
-def pred_maps(image_1, image_2, image_3, image_4):
-    images = [image_1, image_2, image_3, image_4]
+def pred_maps(images):
+    # images = [image_1, image_2, image_3, image_4]
     image_shapes = [image.shape[:2] for image in images]
     images = [Image.fromarray(image) for image in images]
 
@@ -55,9 +55,7 @@ def pred_maps(image_1, image_2, image_3, image_4):
     image_preds = []
     for image, pred in zip(images, preds):
         image_preds.append(
-            # cv2.cvtColor(
-                np.hstack([np.array(image.convert('RGB')), cv2.cvtColor((pred*255).astype(np.uint8), cv2.COLOR_GRAY2RGB)]),
-                # cv2.COLOR_BGR2RGB)
+                np.hstack([np.array(image.convert('RGB')), cv2.cvtColor((pred*255).astype(np.uint8), cv2.COLOR_GRAY2RGB)])
         )
     return image_preds[:]