NHLOCAL commited on
Commit
df7a54d
·
1 Parent(s): 5adb2fc

ניסיון שינוי

Browse files
Files changed (1) hide show
  1. backend.py +1 -11
backend.py CHANGED
@@ -161,7 +161,7 @@ from sam2.sam2_image_predictor import SAM2ImagePredictor
161
 
162
  # נתיבים יחסיים ל-Space של Hugging Face
163
  SAM2_CHECKPOINT = "checkpoints/sam2.1_hiera_tiny.pt"
164
- MODEL_CFG = "sam2.1_hiera_t.yaml"
165
 
166
  sam2_predictor = None # אתחול כ-None
167
  device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -170,16 +170,6 @@ def load_sam2_model():
170
  """טוען את מודל SAM2 באופן גלובלי."""
171
  global sam2_predictor
172
  try:
173
- # הדפסת נתיבים מלאים לבדיקה
174
- print(f"[DEBUG] SAM2_CHECKPOINT (full path): {os.path.abspath(SAM2_CHECKPOINT)}")
175
- print(f"[DEBUG] MODEL_CFG (full path): {os.path.abspath(MODEL_CFG)}")
176
-
177
- # בדיקת קיום הקבצים
178
- if not os.path.exists(SAM2_CHECKPOINT):
179
- raise FileNotFoundError(f"SAM2 checkpoint file not found at: {SAM2_CHECKPOINT}")
180
- if not os.path.exists(MODEL_CFG):
181
- raise FileNotFoundError(f"SAM2 config file not found at: {MODEL_CFG}")
182
-
183
  # טעינת המודל
184
  sam2_model = build_sam2(MODEL_CFG, SAM2_CHECKPOINT, device=device)
185
  sam2_predictor = SAM2ImagePredictor(sam2_model)
 
161
 
162
  # נתיבים יחסיים ל-Space של Hugging Face
163
  SAM2_CHECKPOINT = "checkpoints/sam2.1_hiera_tiny.pt"
164
+ MODEL_CFG = "configs/sam2.1/sam2.1_hiera_t.yaml"
165
 
166
  sam2_predictor = None # אתחול כ-None
167
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
170
  """טוען את מודל SAM2 באופן גלובלי."""
171
  global sam2_predictor
172
  try:
 
 
 
 
 
 
 
 
 
 
173
  # טעינת המודל
174
  sam2_model = build_sam2(MODEL_CFG, SAM2_CHECKPOINT, device=device)
175
  sam2_predictor = SAM2ImagePredictor(sam2_model)