Update Dockerfile

Dockerfile

@@ -1,15 +1,11 @@
 # Use the base image from your Docker container
 FROM trcoot/cpu-casuallm:latest
 
-# Install any additional dependencies or files needed
-# RUN pip install -r requirements.txt
-
-# Set the environment variable for the cache directory
-ENV TRANSFORMERS_CACHE=/app/.cache
+# 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
 
-
 # Start the application (modify this based on your app's entry point)
 CMD ["python", "app.py"]