Spaces:
Running
on
Zero
Running
on
Zero
Update sam2_mask.py
Browse files- sam2_mask.py +5 -1
sam2_mask.py
CHANGED
@@ -153,7 +153,8 @@ def create_sam2_tab():
|
|
153 |
with gr.Column():
|
154 |
with gr.Row():
|
155 |
with gr.Column():
|
156 |
-
sam_input_image = gr.Image(label="input image", interactive=False, type="filepath", visible=False)
|
|
|
157 |
points_map = gr.Image(
|
158 |
label="points map",
|
159 |
type="filepath",
|
@@ -167,7 +168,10 @@ def create_sam2_tab():
|
|
167 |
with gr.Column():
|
168 |
output_image = gr.Image("Segmented Output")
|
169 |
output_result_mask = gr.Image()
|
|
|
|
|
170 |
|
|
|
171 |
# Event handlers
|
172 |
points_map.upload(
|
173 |
fn = preprocess_image,
|
|
|
153 |
with gr.Column():
|
154 |
with gr.Row():
|
155 |
with gr.Column():
|
156 |
+
sam_input_image = gr.Image(label="input image", interactive=False, type="filepath", visible=False)
|
157 |
+
img_prompter = ImagePrompter(show_label=False)
|
158 |
points_map = gr.Image(
|
159 |
label="points map",
|
160 |
type="filepath",
|
|
|
168 |
with gr.Column():
|
169 |
output_image = gr.Image("Segmented Output")
|
170 |
output_result_mask = gr.Image()
|
171 |
+
prompted_output = gr.Image(show_label=False)
|
172 |
+
prompted_data = gr.Dataframe(label="Points")
|
173 |
|
174 |
+
# lambda prompts: (prompts["image"], prompts["points"]),
|
175 |
# Event handlers
|
176 |
points_map.upload(
|
177 |
fn = preprocess_image,
|