Update app.py

app.py

@@ -89,7 +89,7 @@ def get_mask(img, num_people_keep, dilate_kernel_size):
         full_mask = full_mask + blank_mask*instance_mask.to("cpu").numpy()
     
     full_mask = full_mask.reshape((img.shape[0],img.shape[1],1))
-    mask = (cv2.cvtColor(full_mask, cv2.COLOR_GRAY2RGBA)).astype(np.uint8)
+    mask = full_mask.astype(np.uint8)
 
     # Dilation
     kernel = np.ones((dilate_kernel_size, dilate_kernel_size), np.uint8)