download hf cli

Dockerfile

@@ -1,10 +1,10 @@
-# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
-# you will also find guides on how best to write your Dockerfile
-
 FROM python:3.9
 
 WORKDIR /code
 
+# Install huggingface-cli
+RUN pip install --no-cache-dir huggingface_hub
+
 ENV REPO=TheBloke/Llama-2-7B-Chat-GGUF
 ENV MODEL_NAME=llama-2-7b-chat.Q5_K_M.gguf
 
@@ -14,10 +14,11 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY . .
 
+# Download the model
 RUN huggingface-cli download \
-   ${REPO} \
-   ${MODEL_NAME}\
-  --local-dir . \
-  --local-dir-use-symlinks False
+    ${REPO} \
+    ${MODEL_NAME} \
+    --local-dir . \
+    --local-dir-use-symlinks False
 
-CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]