Ivan Shelonik commited on
Commit
165b779
·
1 Parent(s): a87ccd3

upd: transformers cache

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -16,6 +16,7 @@ EXPOSE 5000
16
 
17
  # Set the environment variable for Flask
18
  ENV FLASK_APP=api_server.py
 
19
 
20
  # Run the Flask application
21
  CMD ["flask", "run", "--host=0.0.0.0"]
 
16
 
17
  # Set the environment variable for Flask
18
  ENV FLASK_APP=api_server.py
19
+ ENV TRANSFORMERS_CACHE=$(pwd)/.transformers_cache
20
 
21
  # Run the Flask application
22
  CMD ["flask", "run", "--host=0.0.0.0"]