update

app.py

@@ -9,6 +9,10 @@ from PIL import Image
 import os
 import subprocess
 
+os.makedirs("./ckpt", exist_ok=True)
+# download ViT-H SAM model into ./ckpt
+subprocess.call(["wget", "-q", "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth", "-O", "./ckpt/sam_vit_h_4b8939.pth"])
+
 # def install_cuda_toolkit():
 #     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
 #     CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"