update app.py

app.py

@@ -14,8 +14,8 @@ if os.environ.get('IS_MY_DEBUG') is None:
     result = subprocess.run(['pip', 'install', '-e', 'GroundingDINO'], check=True)
     print(f'pip install GroundingDINO = {result}')
 
-result = subprocess.run(['pip', 'list'], check=True)
-print(f'pip list = {result}')
+# result = subprocess.run(['pip', 'list'], check=True)
+# print(f'pip list = {result}')
 
 sys.path.insert(0, './GroundingDINO')
 
@@ -72,6 +72,7 @@ device = 'cpu'
 
 os.makedirs(output_dir, exist_ok=True)
 groundingdino_model = None
+sam_device = None
 sam_model = None
 sam_predictor = None
 sam_mask_generator = None
@@ -256,7 +257,7 @@ def load_groundingdino_model():
 
 def load_sam_model():
     # initialize SAM
-    global sam_model, sam_predictor, sam_mask_generator
+    global sam_model, sam_predictor, sam_mask_generator, sam_device
     logger.info(f"initialize SAM model...")
     sam_device = device
     sam_model = build_sam(checkpoint=sam_checkpoint).to(sam_device)