Update app.py

app.py

@@ -2,18 +2,25 @@ import gradio
 
 
 import matplotlib.pyplot as plt
-from skimage import morphology
+from skimage import morphology,measure
+from skimage.measure import label
 import numpy as np
 import skimage
 
 
 def inference(img):
-  grayscale = skimage.color.rgb2gray(img)
+  out = skimage.color.rgb2gray(img) # gray
   #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 , min_size=2, connectivity=4)
+    
+  #out = morphology.remove_small_objects(out , min_size=2, connectivity=4)
+  #out = morphology.remove_small_holes(out , min_size=2, connectivity=4)
+
   #out = processed
 
+  out = morphology.remove_small_objects(label(edges), 2,) # noise_reduced
+
+
   # black out pixels
   #mask_x, mask_y = np.where(processed == 0)
   #img[mask_x, mask_y, :3] = 0
@@ -22,7 +29,7 @@ def inference(img):
 
 
   #return img
-  return processed
+  return out
 
 # For information on Interfaces, head to https://gradio.app/docs/
 # For user guides, head to https://gradio.app/guides/