Spaces:
Sleeping
Sleeping
ניסיון שינוי
Browse files- 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)
|