Update Dockerfile

Dockerfile

@@ -9,6 +9,18 @@ RUN apt-get update && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
+# Create cache directories with proper permissions
+RUN mkdir -p /tmp/transformers_cache && \
+    mkdir -p /tmp/hf_home && \
+    mkdir -p /tmp/hf_datasets_cache && \
+    mkdir -p /tmp/diffusers_cache && \
+    chmod -R 777 /tmp
+
+# Set environment variables for HF
+ENV HF_HOME=/tmp/hf_home
+ENV TRANSFORMERS_CACHE=/tmp/transformers_cache
+ENV HF_DATASETS_CACHE=/tmp/hf_datasets_cache
+
 # Copy requirements and install
 COPY requirements.txt .
 RUN pip3 install --no-cache-dir -r requirements.txt
@@ -16,11 +28,9 @@ RUN pip3 install --no-cache-dir -r requirements.txt
 # Copy app files
 COPY app.py .
 
-# Set environment variables for HF
-ENV HF_HOME=/code
-
 # Expose port
 EXPOSE 7860
 
 # Run the application with proper host to expose externally
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+