fixes

OpenAdaptAI · Feb 6, 2024 · 740068e · 740068e
1 parent 7b5517c
commit 740068e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/demo_gpt4v_som.py b/demo_gpt4v_som.py
@@ -178,8 +178,6 @@ class ImageMask(gr.templates.Sketchpad):
     Sets: source="canvas", tool="sketch"
     """
 
-    is_template = True
-
     def __init__(self, **kwargs):
         super().__init__(sources=["upload"], interactive=True, **kwargs)
 
@@ -191,7 +189,7 @@ def preprocess(self, x):
 '''
 
 demo = gr.Blocks()
-image = ImageMask(label="Input", type="pil", brush_radius=20.0, brush_color="#FFFFFF", height=512)
+image = ImageMask(label="Input", type="pil", height=512)
 # image = gr.Image(label="Input", type="pil", height=512)
 slider = gr.Slider(1, 3, value=1.8, label="Granularity") # info="Choose in [1, 1.5), [1.5, 2.5), [2.5, 3] for [seem, semantic-sam (multi-level), sam]"
 mode = gr.Radio(['Automatic', 'Interactive', ], value='Automatic', label="Segmentation Mode")

diff --git a/demo_som.py b/demo_som.py
@@ -126,8 +126,6 @@ class ImageMask(gr.templates.Sketchpad):
     Sets: source="canvas", tool="sketch"
     """
 
-    is_template = True
-
     def __init__(self, **kwargs):
         super().__init__(sources=["upload"], interactive=True, **kwargs)
 
@@ -139,7 +137,7 @@ def preprocess(self, x):
 '''
 
 demo = gr.Blocks()
-image = ImageMask(label="Input", type="pil", brush_radius=20.0, brush_color="#FFFFFF")
+image = ImageMask(label="Input", type="pil")
 slider = gr.Slider(1, 3, value=2, label="Granularity", info="Choose in [1, 1.5), [1.5, 2.5), [2.5, 3] for [seem, semantic-sam (multi-level), sam]")
 mode = gr.Radio(['Automatic', 'Interactive', ], value='Automatic', label="Segmentation Mode")
 image_out = gr.Image(label="Auto generation",type="pil")