docker

Dockerfile

@@ -11,8 +11,9 @@ RUN apt-get update && apt-get install ffmpeg libsm6 libxext6 -y
 # Create the cache directory and set permissions
 RUN mkdir -p /code/transformers_cache && chmod -R 777 /code/transformers_cache
 
-# Set the TRANSFORMERS_CACHE environment variable
+# Set the TRANSFORMERS_CACHE and HF_HOME environment variables
 ENV TRANSFORMERS_CACHE=/code/transformers_cache
+ENV HF_HOME=/code/transformers_cache
 
 COPY . .