Update app.py

app.py

@@ -38,12 +38,18 @@ def visualize_model_output(prediction, img):
         output[:,:,1][prediction[:,:,0]==unq_class] = rgb_class_unique[1]
         output[:,:,2][prediction[:,:,0]==unq_class] = rgb_class_unique[2]
 
-    output = output.astype(np.float64)
-    img = img.astype(np.float64)
-
-    overlayed_image = cv2.addWeighted(img,0.4,output,0.1,0)
-
-    return overlayed_image.astype(np.uint8)
+    output = output.astype(np.int8)
+    img = img.astype(np.int8)
+    
+    bg_img = img.copy()
+    shapes = np.zeros_like(img, np.uint8)
+    shapes = output
+    mask = shapes.astype(bool)
+    alpha = 0.2
+    bg_img[mask] = cv2.addWeighted(bg_img, 1 - alpha, shapes, alpha, 0)[mask]
+    
+
+    return bg_img
 
 
 def do_prediction(model_name, img):