Update Dockerfile

Dockerfile

@@ -2,14 +2,17 @@ FROM python:3.9
 
 WORKDIR /code
 
+# Set the TRANSFORMERS_CACHE environment variable to a writable directory
+ENV TRANSFORMERS_CACHE=/code/cache
+
+# Create the cache directory with appropriate permissions
+RUN mkdir -p /code/cache
+RUN chmod -R 777 /code/cache
+
 COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY . .
 
-# Create the model directory with appropriate permissions
-RUN mkdir -p /code/model
-RUN chmod -R 777 /code/model
-
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]