Update app.py

app.py

@@ -63,7 +63,8 @@ def gen_pred(img, model=seg_model):
     img = img/255
     img = tf.expand_dims(img, axis=0)
     pred = model.predict(img)
-    pred = np.squeeze(pred, axis=0)
+    # pred = np.squeeze(pred, axis=0)
+    
     # img_bytes = pred.tobytes()
     # nparr = np.frombuffer(img_bytes, np.byte)
     # pred_pil = cv2.imdecode(nparr, cv2.IMREAD_REDUCED_COLOR_8)