test

Dockerfile

@@ -13,6 +13,7 @@ RUN mkdir -p /code/transformers_cache
 
 # Set the TRANSFORMERS_CACHE environment variable
 ENV TRANSFORMERS_CACHE=/code/transformers_cache
+ENV HF_HOME=/code/transformers_cache
 
 COPY . .
 

main.py

@@ -5,6 +5,7 @@ from transformers import pipeline
 
 # Set the cache directory to a location with write permissions
 os.environ['TRANSFORMERS_CACHE'] = './transformers_cache'
+os.environ['HF_HOME'] = './transformers_cache'
 
 app = Flask(__name__)
 CORS(app)