Spaces:
Sleeping
Sleeping
IZERE HIRWA Roger
commited on
Commit
·
cb43dd3
1
Parent(s):
0ac79d5
pp
Browse files- Dockerfile +6 -1
- app.py +1 -1
Dockerfile
CHANGED
|
@@ -10,7 +10,9 @@ ENV DEBIAN_FRONTEND=noninteractive \
|
|
| 10 |
# ↓ Hugging Face cache inside the container (optional)
|
| 11 |
HF_HOME=/opt/hf_cache \
|
| 12 |
# Set MPLCONFIGDIR to a writable directory
|
| 13 |
-
MPLCONFIGDIR=/tmp/matplotlib-cache
|
|
|
|
|
|
|
| 14 |
|
| 15 |
# ––– OS packages –––
|
| 16 |
RUN apt-get update && \
|
|
@@ -43,5 +45,8 @@ RUN mkdir -p weights && \
|
|
| 43 |
wget -q -O weights/groundingdino_swint_ogc.pth \
|
| 44 |
https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth
|
| 45 |
|
|
|
|
|
|
|
|
|
|
| 46 |
EXPOSE 7860
|
| 47 |
ENTRYPOINT ["python", "app.py", "--host", "0.0.0.0", "--port", "7860"]
|
|
|
|
| 10 |
# ↓ Hugging Face cache inside the container (optional)
|
| 11 |
HF_HOME=/opt/hf_cache \
|
| 12 |
# Set MPLCONFIGDIR to a writable directory
|
| 13 |
+
MPLCONFIGDIR=/tmp/matplotlib-cache \
|
| 14 |
+
# Set Fontconfig cache directory to a writable location
|
| 15 |
+
FONTCONFIG_PATH=/tmp/fontconfig-cache
|
| 16 |
|
| 17 |
# ––– OS packages –––
|
| 18 |
RUN apt-get update && \
|
|
|
|
| 45 |
wget -q -O weights/groundingdino_swint_ogc.pth \
|
| 46 |
https://github.com/IDEA-Research/GroundingDINO/releases/download/v0.1.0-alpha/groundingdino_swint_ogc.pth
|
| 47 |
|
| 48 |
+
# Set permissions for the working directory
|
| 49 |
+
RUN chmod 777 /workspace
|
| 50 |
+
|
| 51 |
EXPOSE 7860
|
| 52 |
ENTRYPOINT ["python", "app.py", "--host", "0.0.0.0", "--port", "7860"]
|
app.py
CHANGED
|
@@ -28,7 +28,7 @@ device = torch.device("cpu")
|
|
| 28 |
DINO_CKPT = "weights/groundingdino_swint_ogc.pth"
|
| 29 |
SAM_CKPT = "weights/sam_vit_h_4b8939.pth"
|
| 30 |
|
| 31 |
-
grounder = GroundingModel(DINO_CKPT, device=device)
|
| 32 |
sam = sam_model_registry["vit_h"](checkpoint=SAM_CKPT).to(device)
|
| 33 |
predictor = SamPredictor(sam)
|
| 34 |
|
|
|
|
| 28 |
DINO_CKPT = "weights/groundingdino_swint_ogc.pth"
|
| 29 |
SAM_CKPT = "weights/sam_vit_h_4b8939.pth"
|
| 30 |
|
| 31 |
+
grounder = GroundingModel(model_checkpoint_path=DINO_CKPT, device=device)
|
| 32 |
sam = sam_model_registry["vit_h"](checkpoint=SAM_CKPT).to(device)
|
| 33 |
predictor = SamPredictor(sam)
|
| 34 |
|