Update app.py

app.py

@@ -82,8 +82,8 @@ def get_mask(img, num_people_keep, dilate_kernel_size):
     eliminate_instances = people_instances[num_people_keep:]
 
     # Generate mask
-    blank_mask = np.ones((image.shape[0],img.shape[1]), dtype=np.uint8) * 255
-    full_mask = np.zeros((image.shape[0],img.shape[1]), dtype=np.uint8) * 255
+    blank_mask = np.ones((img.shape[0],img.shape[1]), dtype=np.uint8) * 255
+    full_mask = np.zeros((img.shape[0],img.shape[1]), dtype=np.uint8) * 255
 
     for instance_mask in eliminate_instances.pred_masks:
         full_mask = full_mask + blank_mask*instance_mask.to("cpu").numpy()