update app.py

app.py

@@ -250,11 +250,13 @@ def set_device():
 
 def load_groundingdino_model():
     # initialize groundingdino model
+    global groundingdino_model
     logger.info(f"initialize groundingdino model...")
     groundingdino_model = load_model_hf(config_file, ckpt_repo_id, ckpt_filenmae)
 
 def load_sam_model():
     # initialize SAM
+    global sam_model, sam_predictor, sam_mask_generator
     logger.info(f"initialize SAM model...")
     sam_device = device
     sam_model = build_sam(checkpoint=sam_checkpoint).to(sam_device)
@@ -263,6 +265,7 @@ def load_sam_model():
 
 def load_sd_model():
     # initialize stable-diffusion-inpainting
+    global sd_pipe
     logger.info(f"initialize stable-diffusion-inpainting...")
     sd_pipe = None
     if os.environ.get('IS_MY_DEBUG') is None:
@@ -276,6 +279,7 @@ def load_sd_model():
 
 def load_lama_cleaner_model():
     # initialize lama_cleaner
+    global lama_cleaner_model
     logger.info(f"initialize lama_cleaner...")
     from lama_cleaner.helper import (
         load_img,
@@ -359,6 +363,7 @@ class Ram_Predictor(RamPredictor):
 
 def load_ram_model():
     # load ram model
+    global ram_model
     model_path = "./checkpoints/ram_epoch12.pth"
     ram_config = dict(
         model=dict(
@@ -674,23 +679,22 @@ def change_radio_display(task_type, mask_source_radio):
         num_relation_visible = True
     return  gr.Textbox.update(visible=text_prompt_visible), gr.Textbox.update(visible=inpaint_prompt_visible), gr.Radio.update(visible=mask_source_radio_visible), gr.Slider.update(visible=num_relation_visible) 
 
-set_device()
-load_groundingdino_model()
-load_sam_model()
-load_sd_model()
-load_lama_cleaner_model()
-load_ram_model()
-
-os.system("pip list")
-
 if __name__ == "__main__":
     parser = argparse.ArgumentParser("Grounded SAM demo", add_help=True)
     parser.add_argument("--debug", action="store_true", help="using debug mode")
     parser.add_argument("--share", action="store_true", help="share the app")
     args = parser.parse_args()
-
     print(f'args = {args}')
 
+    set_device()
+    load_groundingdino_model()
+    load_sam_model()
+    load_sd_model()
+    load_lama_cleaner_model()
+    load_ram_model()
+
+    os.system("pip list")
+
     block = gr.Blocks().queue()
     with block:
         with gr.Row():