uzzam2121 commited on
Commit
9ea8d9d
·
1 Parent(s): f1a8bd1

Fix model dir to /home/user/model

Browse files
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=/app/repo/model
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", "/app/repo/model"))
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.")