u

app.py

@@ -25,7 +25,7 @@ def resize_and_show(image):
   cv2.waitKey(1000)
   cv2.destroyAllWindows()
 
-def segmentate(image):
+def segmentate(filepath):
     BG_COLOR = (192, 192, 192) # gray
     MASK_COLOR = (255, 255, 255) # white
 
@@ -39,8 +39,8 @@ def segmentate(image):
       # Loop through demo image(s)
 
         # Create the MediaPipe image file that will be segmented
-        # image = mp.Image.create_from_file("examples/kawamura_sample.jpeg")
-
+        image = mp.Image.create_from_file(filepath)
+        
         # Retrieve the masks for the segmented image
         segmentation_result = segmenter.segment(image)
         category_mask = segmentation_result.category_mask
@@ -67,7 +67,7 @@ examples = [[f'examples/{name}', 3] for name in sorted(os.listdir('examples'))]
 iface = gr.Interface(
     fn=segmentate,
     inputs=[
-        gr.Image(type='numpy', label='Input Image')
+        gr.Image(type='filepath', label='Input Image')
     ],
     outputs=[
         gr.Image(label='image segmentated')