Update app.py

app.py

@@ -18,8 +18,9 @@ def inference(img):
 
   #out = processed
   #edges = get_edges(img.copy())
-  edges = feature.canny(out, sigma=3) # edge detect via canny with sigma 3 
-  out = morphology.remove_small_objects(label(edges), 2,) # noise_reduced
+  #edges = feature.canny(out, sigma=3) # edge detect via canny with sigma 3 
+  #out = morphology.remove_small_objects(label(edges), 2,) # noise_reduced
+  out = morphology.remove_small_objects( img , 2,) # noise_reduced
 
 
   # black out pixels
@@ -30,7 +31,7 @@ def inference(img):
 
 
   #return img
-  return edges
+  return out
 
 # For information on Interfaces, head to https://gradio.app/docs/
 # For user guides, head to https://gradio.app/guides/