Update Dockerfile

Dockerfile

@@ -13,10 +13,10 @@ ENV FLASK_APP=app.main
 ENV FLASK_RUN_HOST=0.0.0.0
 ENV FLASK_RUN_PORT=7860
 
-# Set the Transformers cache directory within the container
-ENV TRANSFORMERS_CACHE=/code/.transformers_cache
+# Set the Hugging Face cache directory within the container
+ENV HF_HOME=/code/.huggingface
 
-# Create the directory for Transformers cache
-RUN mkdir -p /code/.transformers_cache
+# Create the directory for Hugging Face cache and ensure proper permissions
+RUN mkdir -p /code/.huggingface && chmod -R 777 /code/.huggingface
 
 CMD ["flask", "run"]