Update app.py

app.py

@@ -10,7 +10,7 @@ def resize_image(img_in,input_height,input_width):
 
 def visualize_model_output(prediction, img):
     unique_classes = np.unique(prediction[:,:,0])
-    rgb_colors = {'0' : [0, 0, 0],
+    rgb_colors = {'0' : [255, 255, 255],
                  '1' : [255, 0, 0],
                  '2' : [255, 125, 0],
                  '3' : [255, 0, 125],
@@ -45,7 +45,7 @@ def visualize_model_output(prediction, img):
     shapes = np.zeros_like(img, np.uint8)
     shapes = output
     mask = shapes.astype(bool)
-    alpha = 0.2
+    alpha = 0.5
     bg_img[mask] = cv2.addWeighted(bg_img, 1 - alpha, shapes, alpha, 0)[mask]