Update app.py

app.py

@@ -131,6 +131,8 @@ with gr.Blocks(css=css, title='Fast Segment Anything') as demo:
 
     segment_btn_e.click(segment_everything,
                         inputs=[
+                            model,
+                            device,
                             cond_img_e,
                             input_size_slider,
                             iou_threshold,
@@ -176,7 +178,7 @@ with gr.Blocks(css=css, title='Fast Segment Anything') as demo:
     cond_img_p.select(get_points_with_draw, [cond_img_p, add_or_remove], cond_img_p)
 
     segment_btn_p.click(segment_with_points,
-                        inputs=[cond_img_p],
+                        inputs=[model, device, cond_img_p],
                         outputs=[segm_img_p, cond_img_p])
 
     with gr.Tab("Text mode"):
@@ -234,6 +236,8 @@ with gr.Blocks(css=css, title='Fast Segment Anything') as demo:
 
     segment_btn_t.click(segment_everything,
                         inputs=[
+                            model,
+                            device,
                             cond_img_t,
                             input_size_slider_t,
                             iou_threshold,