Update app.py

app.py

@@ -11,12 +11,12 @@ def inference(img):
   grayscale = skimage.color.rgb2gray(img)
   #binarized = np.where(grayscale>0.1, 1, 0)
   #processed = morphology.remove_small_objects(grayscale.astype(bool), min_size=33, connectivity=4).astype(int)
-  processed = morphology.remove_small_objects(grayscale.astype(bool), min_size=2, connectivity=4)
+  processed = morphology.remove_small_objects(grayscale , min_size=2, connectivity=4)
   #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)
+  #img[mask_x, mask_y, :3] = 0
   #mask_x, mask_y = np.where(processed == 0)
   #im[mask_x, mask_y, :3] = 0