Update app.py

app.py

@@ -69,8 +69,8 @@ def gen_pred(img, model=seg_model):
     # pil_img = Image.fromarray(pred, 'RGB')
 
     # importing Image from PIL
-    # img_tmp = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
-    im_PIL = Image.fromarray(pred)
+    img_tmp = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
+    im_PIL = Image.fromarray(img_tmp)
     # im = Image.fromarray((pred * 255).astype(np.uint8))
     # img_bytes = pred.tobytes()
     # nparr = np.frombuffer(img_bytes, np.byte)