Update app.py

app.py

@@ -39,8 +39,11 @@ def visualize_model_output(prediction, img):
         output[:,:,2][prediction[:,:,0]==unq_class] = rgb_class_unique[2]
 
     output = output.astype(np.uint8)
-    im_pil_output = Image.fromarray(output)
-    im_pil = Image.fromarray(img)
+    output_color_coverted = cv2.cvtColor(output, cv2.COLOR_BGR2RGB) 
+    im_pil_output = Image.fromarray(output_color_coverted)
+
+    color_coverted = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) 
+    im_pil = Image.fromarray(color_coverted)
 
     im_pil.paste(im_pil_output, (0,0))