Update Dockerfile

Dockerfile

@@ -7,6 +7,10 @@ WORKDIR /code
 # Set environment variables
 ENV TRANSFORMERS_CACHE /code/.cache/huggingface
 
+# Create necessary directories and grant permissions
+RUN mkdir -p /code/.cache/huggingface && \
+    chmod -R 777 /code/.cache/huggingface
+
 # Copy the requirements file into the container at /code
 COPY ./requirements.txt /code/requirements.txt
 
@@ -21,3 +25,4 @@ EXPOSE 7860
 
 # Command to run the Flask application
 CMD ["python", "app.py"]
+