Spaces:
Runtime error
Runtime error
watchtowerss
commited on
Commit
•
6738b38
1
Parent(s):
53a8438
solve multi-user for sam conflict
Browse files
app.py
CHANGED
@@ -161,6 +161,8 @@ def sam_refine(video_state, point_prompt, click_state, interactive_state, evt:gr
|
|
161 |
interactive_state["negative_click_times"] += 1
|
162 |
|
163 |
# prompt for sam model
|
|
|
|
|
164 |
prompt = get_prompt(click_state=click_state, click_input=coordinate)
|
165 |
|
166 |
mask, logit, painted_image = model.first_frame_click(
|
|
|
161 |
interactive_state["negative_click_times"] += 1
|
162 |
|
163 |
# prompt for sam model
|
164 |
+
model.samcontroler.sam_controler.reset_image()
|
165 |
+
model.samcontroler.sam_controler.set_image(video_state["origin_images"][video_state["select_frame_number"]])
|
166 |
prompt = get_prompt(click_state=click_state, click_input=coordinate)
|
167 |
|
168 |
mask, logit, painted_image = model.first_frame_click(
|