Update app.py

app.py

@@ -13,21 +13,7 @@ from per_segment_anything import sam_model_registry, SamPredictor
 parser = argparse.ArgumentParser()
 parser.add_argument("-op", "--output-path", type=str, default='default')
 args = parser.parse_args()
-
-
-class ImageMask(gr.components.Image):
-    """
-    Sets: source="canvas", tool="sketch"
-    """
-
-    is_template = True
-
-    def __init__(self, **kwargs):
-        super().__init__(source="upload", tool='select', interactive=True, **kwargs)
-
-    def preprocess(self, x):
-        return super().preprocess(x)
-    
+   
 
 def point_selection(mask_sim, topk=1):
     # Top-1 point selection
@@ -150,10 +136,10 @@ def inference_scribble(image):
             Image.fromarray((mask_colors ).astype('uint8'), 'RGB')]
 
 demo = gr.Interface(fn=inference_scribble,
-    inputs=gr.ImageMask(label="[Stroke] Draw on Image", type='pil'),
+    inputs=gr.Image(label="[Stroke] Draw on Image", tool='sketch',type='pil'),
     outputs=[
-        gr.outputs.Image(type="pil", label="Mask with Image"),
-        gr.outputs.Image(type="pil", label="Mask")
+        gr.Image(type="pil", label="Mask with Image"),
+        gr.Image(type="pil", label="Mask")
         ],
     title="SAM based Segment Annotator.",
     description='Sketch the portion where you want to create Mask.',