Nirav Madhani commited on
Commit
db007c9
·
1 Parent(s): 0558e79

Modify cache

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -19,7 +19,7 @@ RUN pip3 install scipy==1.10.1
19
 
20
  # Install FastAPI and Uvicorn for the API
21
  RUN pip3 install fastapi uvicorn
22
-
23
  # Copy and run the model initialization script to cache the model
24
  COPY init_model.py /app/octo
25
  RUN python init_model.py
 
19
 
20
  # Install FastAPI and Uvicorn for the API
21
  RUN pip3 install fastapi uvicorn
22
+ ENV HF_HOME=/app/octo/.cache
23
  # Copy and run the model initialization script to cache the model
24
  COPY init_model.py /app/octo
25
  RUN python init_model.py