Update app.py

app.py

@@ -1,11 +1,16 @@
 import gradio
+import numpy as np
 
 
-import matplotlib.pyplot as plt
-from skimage import morphology,measure,feature
-from skimage.measure import label
-import numpy as np
-import skimage
+#import matplotlib.pyplot as plt
+#from skimage import morphology,measure,feature
+#from skimage.measure import label
+
+#import skimage
+
+from skimage import restoration
+from skimage.filters import threshold_otsu, rank
+from skimage.morphology import closing, square, disk
 
 
 def inference(img):
@@ -29,8 +34,11 @@ def inference(img):
   #mask_x, mask_y = np.where(processed == 0)
   #im[mask_x, mask_y, :3] = 0
 
+  denoise = restoration.denoise_tv_chambolle( out , weight=0.1)
+  thresh = threshold_otsu(denoise)
+  out = closing(denoise  > thresh, square(2))
+
 
-  #return img
   return out
 
 # For information on Interfaces, head to https://gradio.app/docs/