Update Dockerfile

Dockerfile

@@ -1,14 +1,23 @@
 FROM python:3.10-slim
 
+# Install dependencies
 RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
 
-RUN pip install --no-cache-dir torch transformers flask kernels accelerate
+# Install Python packages
+RUN pip install --no-cache-dir torch transformers flask kernels accelerate huggingface_hub
 
-RUN pip install transformers huggingface_hub
+# Create cache dir for Hugging Face models
 RUN mkdir -p /app/model_cache
+ENV HF_HOME=/app/model_cache
+
+# Download GPT-OSS-20B at build time
+RUN python -c "from huggingface_hub import snapshot_download; snapshot_download('openai/gpt-oss-20b', cache_dir='/app/model_cache')"
+
+# Copy app code
 WORKDIR /app
 COPY app.py /app
 RUN chmod -R 777 /app
-ENV HF_HOME=/app/model_cache
+
+# Expose port and start app
 EXPOSE 7860
 CMD ["python", "app.py"]