Upload app.py

app.py

@@ -116,10 +116,11 @@ def put_mask(image,mask,color_rgb=None,border_mask=False,color_temp='jet',num_c=
 
 
     
-    result = overlay_mask(to_pil_image(image.astype(np.uint8)), to_pil_image(mask), colormap = color_temp, alpha=0.4)
-
+    heatmap = cv2.applyColorMap((mask * 255).astype(np.uint8), cv2.COLORMAP_JET)
+    result = cv2.addWeighted(image, 0.6, heatmap, 0.4, 0)
   
     return np.array(result)
+  
 
 
 def visualize_geometry_prior(RGB_path, Depth_path, index_list=[[584]], cmap_list = ['jet_r'],x=0,y=0):