Update app.py

app.py

@@ -66,9 +66,10 @@ def gen_pred(img, model=seg_model):
     print(pred)
     img_bytes = pred.tobytes()
     nparr = np.frombuffer(img_bytes, np.byte)
-    # pred_pil = cv2.imdecode(nparr, cv2.IMREAD_REDUCED_COLOR_8)
+    pred_pil = cv2.imdecode(nparr, cv2.IMREAD_REDUCED_COLOR_8)
+    print(pred_pil)
     # return "UI in developing process ..."
-    return nparr
+    return pred_pil
 
 title = "<h1 style='text-align: center;'>Semantic Segmentation</h1>"
 description = "Upload an image and get prediction mask"