Update Dockerfile

Dockerfile

@@ -22,6 +22,12 @@ FROM python:3.8-slim
 # Allow statements and log messages to immediately appear in the logs
 ENV PYTHONUNBUFFERED=True
 
+# Set the TRANSFORMERS_CACHE environment variable
+ENV TRANSFORMERS_CACHE=/app/cache
+
+# Create the cache directory
+RUN mkdir -p /app/cache
+
 # Copy requirements.txt and install dependencies
 COPY requirements.txt .
 RUN pip install -r requirements.txt
@@ -33,3 +39,4 @@ WORKDIR /app
 # Run the FastAPI app with uvicorn
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
+