Update Dockerfile

Dockerfile

@@ -2,13 +2,16 @@ FROM python:3.9
 
 WORKDIR /code
 
+# Create the cache directory
+RUN mkdir -p /code/hf_cache
+
+# Set the HF_HOME environment variable to the cache directory
+ENV HF_HOME=/code/hf_cache
+
 COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY ./app.py /code/app.py
 
-# Ensure /code has the correct permissions
-RUN chmod -R 777 /code
-
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]