upd: transformers cache

Dockerfile

@@ -16,7 +16,7 @@ EXPOSE 5000
 
 # Set the environment variable for Flask
 ENV FLASK_APP=api_server.py
-ENV TRANSFORMERS_CACHE=transformers_cache
+# ENV TRANSFORMERS_CACHE=transformers_cache
 
 # Run the Flask application
 CMD ["flask", "run", "--host=0.0.0.0"]

api_server.py

@@ -2,7 +2,10 @@ import os
 import time
 import numpy as np
 
-os.environ['TRANSFORMERS_CACHE'] = './artifacts/'
+from pathlib import Path
+
+os.environ['TRANSFORMERS_CACHE'] = Path('./artifacts/').absolute()
+
 
 # Disable tensorflow warnings
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'