p

Dockerfile

@@ -48,5 +48,7 @@ RUN mkdir -p weights && \
 # Set permissions for the working directory
 RUN chmod 777 /workspace
 
+RUN mkdir -p /tmp/fontconfig-cache && chmod 777 /tmp/fontconfig-cache
+
 EXPOSE 7860
 ENTRYPOINT ["python", "app.py", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -25,10 +25,11 @@ from segment_anything import sam_model_registry, SamPredictor
 # ─── Load models once ───────────────────────────────────────────────────────────
 device = torch.device("cpu")
 
+DINO_CONFIG = "GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py"
 DINO_CKPT = "weights/groundingdino_swint_ogc.pth"
 SAM_CKPT  = "weights/sam_vit_h_4b8939.pth"
 
-grounder = GroundingModel(model_checkpoint_path=DINO_CKPT, device=device)
+grounder = GroundingModel(model_config_path=DINO_CONFIG, model_checkpoint_path=DINO_CKPT, device=device)
 sam = sam_model_registry["vit_h"](checkpoint=SAM_CKPT).to(device)
 predictor = SamPredictor(sam)