docker update

Dockerfile

@@ -5,6 +5,11 @@ ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
 ENV DEBIAN_FRONTEND=noninteractive
 
+# Hugging Face cache locations (feel free to rename as needed)
+ENV TRANSFORMERS_CACHE=/app/.cache
+ENV HF_HOME=/app/.cache
+ENV HF_HUB_CACHE=/app/.cache
+
 # Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     python3 \
@@ -15,6 +20,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
+# Create the cache directory and fix permissions
+RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
+
 # Set the working directory
 WORKDIR /app
 
@@ -29,4 +37,4 @@ RUN pip3 install --no-cache-dir -r requirements.txt
 COPY app.py .
 
 # Set the default command to run the application
-CMD ["python3", "app.py"]
+CMD ["python3", "app.py"]