watchtowerss commited on
Commit
6738b38
1 Parent(s): 53a8438

solve multi-user for sam conflict

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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(