.................

Dockerfile

@@ -14,8 +14,8 @@ COPY ./core ./core
 # Set the PYTHONPATH to include the /app directory
 ENV PYTHONPATH="/app"
 
-# Set the TRANSFORMERS_CACHE to a writable directory
-ENV TRANSFORMERS_CACHE="/app/cache"
+# Set the HF_HOME to a writable directory
+ENV HF_HOME="/app/cache"
 
 # Create the cache directory
 RUN mkdir -p /app/cache
@@ -47,8 +47,8 @@ COPY --from=core /app/core/engine.pickle /app/api/engine.pickle
 # Set the PYTHONPATH to include the /app directory
 ENV PYTHONPATH="/app"
 
-# Set the TRANSFORMERS_CACHE to a writable directory
-ENV TRANSFORMERS_CACHE="/app/cache"
+# Set the HF_HOME to a writable directory
+ENV HF_HOME="/app/cache"
 
 # Create the cache directory
 RUN mkdir -p /app/cache