Update Dockerfile

Dockerfile

@@ -13,5 +13,10 @@ ENV FLASK_APP=app.main
 ENV FLASK_RUN_HOST=0.0.0.0
 ENV FLASK_RUN_PORT=7860
 
-# Command to run the Flask application
+# Set the Transformers cache directory within the container
+ENV TRANSFORMERS_CACHE=/code/.transformers_cache
+
+# Create the directory for Transformers cache
+RUN mkdir -p /code/.transformers_cache
+
 CMD ["flask", "run"]