Update app.py

app.py

@@ -9,8 +9,8 @@ import skimage
 
 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=33, connectivity=4).astype(int)
+  #binarized = np.where(grayscale>0.1, 1, 0)
+  processed = morphology.remove_small_objects(grayscale.astype(bool), min_size=33, connectivity=4).astype(int)
   #out = processed
 
   # black out pixels
@@ -20,7 +20,7 @@ def inference(img):
   #im[mask_x, mask_y, :3] = 0
 
 
-  return binarized
+  return processed
 
 # For information on Interfaces, head to https://gradio.app/docs/
 # For user guides, head to https://gradio.app/guides/