Spaces:
Runtime error
Runtime error
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(
|