Update Dockerfile

Dockerfile

@@ -1,7 +1,12 @@
-
-
 # Use the base image from your Docker container
 FROM trcoot/cpu-casuallm:latest
 
+# Set the environment variable for the Hugging Face cache directory
+ENV HF_HOME=/app/.cache 
+
+# Create the cache directory and give the appropriate permissions
+RUN mkdir -p /app/.cache && chmod 777 /app/.cache
+RUN mkdir -p /app/gradio_flagged && chmod 777 /app/gradio_flagged
+
 # Start the application (modify this based on your app's entry point)
-CMD ["python", "app.py"] 
+CMD ["python", "app.py"]