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

Dockerfile

@@ -14,6 +14,12 @@ 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"
+
+# Create the cache directory
+RUN mkdir -p /app/cache
+
 # Run the initialization script to load and serialize the dataset
 RUN python ./core/initialization.py
 
@@ -41,6 +47,12 @@ 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"
+
+# Create the cache directory
+RUN mkdir -p /app/cache
+
 # Expose the API port
 EXPOSE 9999