Update app.py

app.py

@@ -58,13 +58,13 @@ def gen_pred(img, model=seg_model):
     # pil_image = img.convert('RGB')
     # open_cv_image = np.array(pil_image)
     # img = open_cv_image[:, :, ::-1].copy() 
-    # # img = cv2.imread("./003e2c95d.jpg")
+    img = cv2.imread("./003e2c95d.jpg")
     img = img[::IMG_SCALING[0], ::IMG_SCALING[1]]
     img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     img = img/255
     img = tf.expand_dims(img, axis=0)
     pred = model.predict(img)
-    pred = np.squeeze(pred, axis=1)
+    pred = np.squeeze(pred, axis=0)
     # image = Image.fromarray(np.uint8(cm.plasma(pred)*255))
     pil_img = Image.fromarray(pred, 'RGB')
     # im = Image.fromarray((pred * 255).astype(np.uint8))