update

sam2edit.py

@@ -58,6 +58,7 @@ def create_demo():
         print('segment_anything not installed')
         result = subprocess.run(['pip', 'install', 'git+https://github.com/facebookresearch/segment-anything.git'], check=True)
         print(f'Install segment_anything {result}')   
+        from segment_anything import sam_model_registry, SamAutomaticMaskGenerator
     if not os.path.exists('./models/sam_vit_h_4b8939.pth'):
         result = subprocess.run(['wget', 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth', '-P', 'models'], check=True)
         print(f'Download sam_vit_h_4b8939.pth {result}')   

sam2image.py

@@ -58,7 +58,8 @@ def create_demo():
     except ImportError:
         print('segment_anything not installed')
         result = subprocess.run(['pip', 'install', 'git+https://github.com/facebookresearch/segment-anything.git'], check=True)
-        print(f'Install segment_anything {result}')   
+        print(f'Install segment_anything {result}')
+        from segment_anything import sam_model_registry, SamAutomaticMaskGenerator        
     if not os.path.exists('./models/sam_vit_h_4b8939.pth'):
         result = subprocess.run(['wget', 'https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth', '-P', 'models'], check=True)
         print(f'Download sam_vit_h_4b8939.pth {result}')