Update app.py

app.py

@@ -70,9 +70,10 @@ def gen_pred(img=inputs, model=seg_model):
     pred = np.squeeze(pred, axis=0)
     print(pred)
     # color_coverted = cv2.cvtColor(pred, cv2.COLOR_BGR2RGB)
-    pil_image = Image.fromarray(pred)
+    # pil_image = Image.fromarray(pred)
+    PIL_image = Image.fromarray(pred.astype('uint8'), 'RGB')
     "UI in developing process ..."
-    return pil_image
+    return PIL_image
 
 
 title = "<h1 style='text-align: center;'>Semantic Segmentation</h1>"