Initialisation 3

Dockerfile

@@ -7,9 +7,6 @@ WORKDIR /app
 # Copier les fichiers nécessaires dans le conteneur
 COPY . /app
 
-# Add segment-anything-2 to PYTHONPATH so `sam2` can be imported
-ENV PYTHONPATH=/app/sam2:$PYTHONPATH
-
 # Installer les dépendances
 RUN pip install --no-cache-dir -r requirements.txt
 
@@ -17,5 +14,4 @@ RUN pip install --no-cache-dir -r requirements.txt
 EXPOSE 7860
 
 # Commande pour démarrer Flask
-CMD ["python", "app.py"]
-
+CMD ["python", "app.py"]

app.py

@@ -86,8 +86,8 @@ load_training_status()
 os.environ["TORCH_CUDNN_SDPA_ENABLED"] = "0"
 
 # Initialize SAM Predictor
-MODEL_CFG = r"C:\codes\sam2\segment-anything-2\sam2\configs\sam2.1\sam2.1_hiera_l.yaml"
-CHECKPOINT = r"C:\codes\sam2\segment-anything-2\checkpoints\sam2.1_hiera_large.pt"
+MODEL_CFG = r"C:\codes\sam2\sam2\configs\sam2.1\sam2.1_hiera_l.yaml"
+CHECKPOINT = r"C:\codes\sam2\checkpoints\sam2.1_hiera_large.pt"
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 predictor = Predictor(MODEL_CFG, CHECKPOINT, DEVICE)