m

handler.py

@@ -126,10 +126,15 @@ sam_model = None
 
 
 def get_sam_vit_h_4b8939():
-    if not os.path.exists('./sam_vit_h_4b8939.pth'):
-        logger.info(f"get sam_vit_h_4b8939.pth...")
-        result = subprocess.run(['wget', 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth'], check=True)
-        print(f'wget sam_vit_h_4b8939.pth result = {result}')
+    url = 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth'
+    file_path = './sam_vit_h_4b8939.pth'
+
+    if not os.path.exists(file_path):
+        logger.info("Downloading sam_vit_h_4b8939.pth...")
+        response = requests.get(url)
+        with open(file_path, 'wb') as f:
+            f.write(response.content)
+        print('Downloaded sam_vit_h_4b8939.pth')
 
 get_sam_vit_h_4b8939()
 logger.info(f"initialize SAM model...")