Pream912 commited on
Commit
3865970
Β·
verified Β·
1 Parent(s): cdfb66d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -129,10 +129,8 @@ _ONES3x3 = np.ones((3, 3), dtype=np.float32)
129
  print(f"[GPU] torch_cuda={_TORCH_CUDA} cupy={_CUPY} cucim={_CUCIM} cv2_cuda={_CV2_CUDA}")
130
 
131
  # ─── SAM HuggingFace endpoint ────────────────────────────────────────────────
132
- HF_REPO = "Pream912/sam"
133
- HF_API = f"https://huggingface.co/{HF_REPO}/resolve/main"
134
  SAM_CKPT = Path(tempfile.gettempdir()) / "sam_vit_h_4b8939.pth"
135
- SAM_URL = f"{HF_API}/sam_vit_h_4b8939.pth"
136
 
137
  DPI = 300
138
  SCALE_FACTOR = 100
 
129
  print(f"[GPU] torch_cuda={_TORCH_CUDA} cupy={_CUPY} cucim={_CUCIM} cv2_cuda={_CV2_CUDA}")
130
 
131
  # ─── SAM HuggingFace endpoint ────────────────────────────────────────────────
 
 
132
  SAM_CKPT = Path(tempfile.gettempdir()) / "sam_vit_h_4b8939.pth"
133
+ SAM_URL = "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth"
134
 
135
  DPI = 300
136
  SCALE_FACTOR = 100