fix

Dockerfile

@@ -1,4 +1,4 @@
-# Torch 1.10 (cu111) + Detectron2 0.6 on CUDA 11.1, lean for HF Spaces
+# Torch 1.10 (cu111) + Detectron2 0.6 on CUDA 11.1
 FROM nvidia/cuda:11.1.1-cudnn8-runtime-ubuntu20.04
 
 ENV DEBIAN_FRONTEND=noninteractive \
@@ -34,6 +34,6 @@ WORKDIR /app
 COPY app.py /app/app.py
 
 # Copy pre-uploaded model(s)
-COPY models /models
+COPY models /weights
 
 CMD ["python3", "app.py"]

app.py

@@ -47,10 +47,10 @@ from detectree2.models.train import setup_cfg
 
 # --------- Config ---------
 WEIGHTS = {
-    # NOTE: Dockerfile downloads this to /models
-    "Forest": "/models/230103_randresize_full.pth",
-    # "Urban": "/models/urban_trees_Cambridge_20230630.pth",
-    # "Flexi": "/models/250312_flexi.pth",
+    # NOTE: Dockerfile downloads this to /weights
+    "Forest": "/weights/230103_randresize_full.pth",
+    # "Urban": "/weights/urban_trees_Cambridge_20230630.pth",
+    # "Flexi": "/weights/250312_flexi.pth",
 }
 DEFAULT_ENV = "Forest"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"