Update Dockerfile

Dockerfile

@@ -1,22 +1,25 @@
-# USE THE OFFICIAL PYTHON 3.9 SLIM IMAGE
-FROM python:3.9-slim
-
-# SET THE WORKING DIRECTORY INSIDE THE CONTAINER
-WORKDIR /code
-
-# COPY THE REQUIREMENTS FILE FIRST FOR EFFICIENT CACHING
-COPY ./requirements.txt /code/requirements.txt
-
-# INSTALL ALL THE REQUIRED LIBRARIES
-RUN pip install --no-cache-dir --upgrade pip -r /code/requirements.txt
-
-# COPY THE REST OF YOUR APPLICATION CODE
-COPY ./app.py /code/app.py
-COPY ./model_handler.py /code/model_handler.py
-
-# HUGGING FACE SPACES RUNS APPS ON PORT 7860
-EXPOSE 7860
-
-# THE COMMAND TO START THE GUNICORN PRODUCTION SERVER
-# IT RUNS THE 'FLASK_APP' OBJECT FROM THE 'app.py' FILE
+# USE THE OFFICIAL PYTHON 3.9 SLIM IMAGE
+FROM python:3.9-slim
+
+# SET THE WORKING DIRECTORY INSIDE THE CONTAINER
+WORKDIR /code
+
+# SET A WRITABLE CACHE DIRECTORY FOR HUGGING FACE MODELS
+ENV HUGGINGFACE_HUB_CACHE=/tmp/.cache
+# --------------------
+
+# COPY THE REQUIREMENTS FILE FIRST FOR EFFICIENT CACHING
+COPY ./requirements.txt /code/requirements.txt
+
+# INSTALL ALL THE REQUIRED LIBRARIES
+RUN pip install --no-cache-dir --upgrade pip -r /code/requirements.txt
+
+# COPY THE REST OF YOUR APPLICATION CODE
+COPY ./app.py /code/app.py
+COPY ./model_handler.py /code/model_handler.py
+
+# HUGGING FACE SPACES RUNS APPS ON PORT 7860
+EXPOSE 7860
+
+# THE COMMAND TO START THE GUNICORN PRODUCTION SERVER
 CMD ["gunicorn", "--bind", "0.0.0.0:7860", "app:FLASK_APP"]