Update Dockerfile

Dockerfile

@@ -1,3 +1,4 @@
+
 FROM python:3.10-slim
 
 # Install system dependencies
@@ -11,6 +12,13 @@ RUN apt-get update && apt-get install -y \
 # Set working directory
 WORKDIR /app
 
+# Create a cache directory with proper permissions
+RUN mkdir -p /app/cache/huggingface && chmod -R 777 /app/cache
+
+# Set environment variables for Hugging Face cache
+ENV HF_HOME=/app/cache/huggingface
+ENV HUGGINGFACE_HUB_CACHE=/app/cache/huggingface/hub
+
 # Copy requirements file
 COPY requirements.txt .
 
@@ -20,11 +28,11 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy application code
 COPY . .
 
+# Download model weights during build (optional, if disk space allows)
+RUN huggingface-cli download openai/gpt-oss-20b --include "original/*" --local-dir /app/gpt-oss-20b || true
+
 # Expose port for the API
 EXPOSE 8000
 
-# Download model weights
-RUN huggingface-cli download openai/gpt-oss-20b --include "original/*" --local-dir /app/gpt-oss-20b
-
 # Command to run the application
 CMD ["python", "app.py"]