Somunia commited on
Commit
0f48a24
1 Parent(s): a0f861b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +7 -0
Dockerfile CHANGED
@@ -4,5 +4,12 @@ FROM trcoot/cpu-casuallm:latest
4
  # Install any additional dependencies or files needed
5
  # RUN pip install -r requirements.txt
6
 
 
 
 
 
 
 
 
7
  # Start the application (modify this based on your app's entry point)
8
  CMD ["python", "app.py"]
 
4
  # Install any additional dependencies or files needed
5
  # RUN pip install -r requirements.txt
6
 
7
+ # Set the environment variable for the cache directory
8
+ ENV TRANSFORMERS_CACHE=/app/.cache
9
+
10
+ # Create the cache directory and give the appropriate permissions
11
+ RUN mkdir -p /app/.cache && chmod 777 /app/.cache
12
+
13
+
14
  # Start the application (modify this based on your app's entry point)
15
  CMD ["python", "app.py"]