Initial commit15323p

app.py

@@ -113,8 +113,9 @@ load_training_status()
 os.environ["TORCH_CUDNN_SDPA_ENABLED"] = "0"
 
 # Initialize SAM Predictor
-MODEL_CFG = r"project/sam2/sam2_hiera_l.yaml"
-CHECKPOINT = r"project/sam2/sam2.1_hiera_large.pt"
+MODEL_CFG = os.path.join(BASE_DIR, "sam2", "sam2_hiera_l.yaml")
+CHECKPOINT = os.path.join(BASE_DIR, "sam2", "sam2.1_hiera_large.pt")
+print(f"Chargement de {MODEL_CFG} et {CHECKPOINT} sur l'appareil .")
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 predictor = Predictor(MODEL_CFG, CHECKPOINT, DEVICE)