JunchuanYu commited on
Commit
ba03bf3
·
1 Parent(s): 00b8625

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -61,8 +61,8 @@ with gr.Blocks(theme='gradio/soft') as demo:
61
  autoseg_button = gr.Button("Auto Segment",variant="primary")
62
  emptyBtn = gr.Button("Restart",variant="secondary")
63
 
64
- interseg_button.click(interactive_seg, inputs=[input_img,x,y,label], outputs=[output_img,mask_img])
65
- autoseg_button.click(auto_seg, inputs=[input_img,thresh,points], outputs=[mask_img])
66
 
67
  clear_bn.click(clear_point,outputs=[gallery,mode,x,y,label],show_progress=True)
68
  emptyBtn.click(reset_state,outputs=[input_img,gallery,output_img,mask_img,thresh,points,mode,x,y,label],show_progress=True,)
 
61
  autoseg_button = gr.Button("Auto Segment",variant="primary")
62
  emptyBtn = gr.Button("Restart",variant="secondary")
63
 
64
+ interseg_button.click(interactive_seg, inputs=[predictor,input_img,x,y,label], outputs=[output_img,mask_img])
65
+ autoseg_button.click(auto_seg, inputs=[sam,input_img,thresh,points], outputs=[mask_img])
66
 
67
  clear_bn.click(clear_point,outputs=[gallery,mode,x,y,label],show_progress=True)
68
  emptyBtn.click(reset_state,outputs=[input_img,gallery,output_img,mask_img,thresh,points,mode,x,y,label],show_progress=True,)