22

Dockerfile

@@ -1,4 +1,3 @@
-
 FROM python:3.9
 
 WORKDIR /app
@@ -9,10 +8,10 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 COPY . .
 
-# Set environment variable for transformers cache
-ENV TRANSFORMERS_CACHE=/app/cache
+# Set environment variable for Hugging Face cache
+ENV HF_HOME=/app/cache
 
-RUN mkdir -p /app/cache
+RUN mkdir -p /app/cache && chmod -R 777 /app/cache
 
 EXPOSE 5000