Update app.py

app.py

@@ -18,8 +18,6 @@ loc =hf_hub_download(repo_id="JunchuanYu/files_for_segmentRS", filename="utils.p
 sys.path.append(loc)
 from utils import *
 
-predictor,sam=prediction(model_type='vit_b')
-
 with gr.Blocks(theme='gradio/soft') as demo:
     gr.Markdown(title)
     with gr.Accordion("Instructions For User 👉", open=False):
@@ -58,11 +56,12 @@ with gr.Blocks(theme='gradio/soft') as demo:
             outputs=[output_img],
             cache_examples=False,examples_per_page=5)
 
+    model_type='vit_b'
     autoseg_button = gr.Button("Auto Segment",variant="primary")
     emptyBtn = gr.Button("Restart",variant="secondary")
 
-    interseg_button.click(interactive_seg, inputs=[predictor,input_img,x,y,label], outputs=[output_img,mask_img])
-    autoseg_button.click(auto_seg, inputs=[sam,input_img,thresh,points], outputs=[mask_img])
+    interseg_button.click(interactive_seg, inputs=[input_img,x,y,label,model_type], outputs=[output_img,mask_img])
+    autoseg_button.click(auto_seg, inputs=[input_img,thresh,points], outputs=[mask_img])
 
     clear_bn.click(clear_point,outputs=[gallery,mode,x,y,label],show_progress=True)
     emptyBtn.click(reset_state,outputs=[input_img,gallery,output_img,mask_img,thresh,points,mode,x,y,label],show_progress=True,)