Update Dockerfile

Dockerfile

@@ -1,26 +1,34 @@
-FROM python:3.9-slim
-
-WORKDIR /code
-
-# Install system dependencies
-RUN apt-get update && apt-get install -y \
-    build-essential \
-    curl \
-    software-properties-common \
-    git \
-    && rm -rf /var/lib/apt/lists/*
-
-# Copy requirements first for better caching
-COPY requirements.txt .
-
-# Install Python dependencies
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Copy the rest of the application
-COPY . .
-
-# Make port 7860 available 
-EXPOSE 7860
-
-# Run the application
+FROM python:3.9-slim
+
+WORKDIR /code
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    curl \
+    software-properties-common \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+# Create cache directory with proper permissions
+RUN mkdir -p /root/.cache/huggingface && \
+    chmod -R 777 /root/.cache
+
+# Set environment variables
+ENV TRANSFORMERS_CACHE=/root/.cache/huggingface
+ENV HF_HOME=/root/.cache/huggingface
+
+# Copy requirements first for better caching
+COPY requirements.txt .
+
+# Install Python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the rest of the application
+COPY . .
+
+# Make port 7860 available 
+EXPOSE 7860
+
+# Run the application
 CMD ["python", "app.py"]