Update app.py

app.py

@@ -11,14 +11,16 @@ def inference(img):
   grayscale = skimage.color.rgb2gray(img)
   binarized = np.where(grayscale>0.1, 1, 0)
   processed = morphology.remove_small_objects(binarized.astype(bool), min_size=2, connectivity=2).astype(int)
-  out = binarized
+  #out = processed
 
   # black out pixels
+  mask_x, mask_y = np.where(processed == 0)
+  img[mask_x, mask_y, :3] = 0
   #mask_x, mask_y = np.where(processed == 0)
   #im[mask_x, mask_y, :3] = 0
 
 
-  return out
+  return img
 
 # For information on Interfaces, head to https://gradio.app/docs/
 # For user guides, head to https://gradio.app/guides/