Spaces:
Runtime error
Runtime error
uzzam2121 commited on
Commit ·
9ea8d9d
1
Parent(s): f1a8bd1
Fix model dir to /home/user/model
Browse files- Dockerfile +4 -2
- src/deploy/huggingface/download_model.py +1 -1
Dockerfile
CHANGED
|
@@ -16,9 +16,11 @@ RUN pip install --no-cache-dir -r /tmp/requirements.txt huggingface_hub \
|
|
| 16 |
|
| 17 |
COPY --chown=user:user src/ /app/repo/
|
| 18 |
|
| 19 |
-
RUN chmod +x /app/repo/deploy/huggingface/entrypoint.sh
|
|
|
|
|
|
|
| 20 |
|
| 21 |
-
ENV MODEL_PATH=/
|
| 22 |
ENV HF_MODEL_REPO=uzzam24/deberta-complexity-lcp
|
| 23 |
ENV APP_ROOT=/app/repo
|
| 24 |
ENV PYTHONPATH=/app/repo:/app/repo/deploy/backend
|
|
|
|
| 16 |
|
| 17 |
COPY --chown=user:user src/ /app/repo/
|
| 18 |
|
| 19 |
+
RUN chmod +x /app/repo/deploy/huggingface/entrypoint.sh \
|
| 20 |
+
&& mkdir -p /home/user/model /home/user/.cache/huggingface \
|
| 21 |
+
&& chown -R user:user /home/user
|
| 22 |
|
| 23 |
+
ENV MODEL_PATH=/home/user/model
|
| 24 |
ENV HF_MODEL_REPO=uzzam24/deberta-complexity-lcp
|
| 25 |
ENV APP_ROOT=/app/repo
|
| 26 |
ENV PYTHONPATH=/app/repo:/app/repo/deploy/backend
|
src/deploy/huggingface/download_model.py
CHANGED
|
@@ -15,7 +15,7 @@ from pathlib import Path
|
|
| 15 |
|
| 16 |
def main() -> None:
|
| 17 |
repo = os.environ.get("HF_MODEL_REPO", "").strip()
|
| 18 |
-
target = Path(os.environ.get("MODEL_PATH", "/
|
| 19 |
|
| 20 |
if not repo:
|
| 21 |
print("HF_MODEL_REPO not set — expecting model_weights.pt already on disk.")
|
|
|
|
| 15 |
|
| 16 |
def main() -> None:
|
| 17 |
repo = os.environ.get("HF_MODEL_REPO", "").strip()
|
| 18 |
+
target = Path(os.environ.get("MODEL_PATH", "/home/user/model"))
|
| 19 |
|
| 20 |
if not repo:
|
| 21 |
print("HF_MODEL_REPO not set — expecting model_weights.pt already on disk.")
|