Update Dockerfile

Dockerfile

@@ -1,7 +1,8 @@
 FROM python:3.9
 
-# Set TRANSFORMERS_CACHE environment variable
-ENV TRANSFORMERS_CACHE=/code/.cache/huggingface/transformers
+# Set HF_HOME environment variable to specify cache directory
+ENV HF_HOME=/code/.cache/huggingface
+
 
 WORKDIR /code
 
@@ -11,4 +12,8 @@ RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY . .
 
+# Set permissions for cache directory
+RUN mkdir -p /code/.cache/huggingface/transformers \
+    && chmod -R 777 /code/.cache/huggingface
+
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]