Update Dockerfile

Dockerfile

@@ -8,7 +8,6 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 
 # Install all the "brains" Maira needs
-# Added flask-cors to the list below!
 RUN pip install --no-cache-dir \
     flask \
     flask-cors \
@@ -19,8 +18,16 @@ RUN pip install --no-cache-dir \
 # Set the working directory
 WORKDIR /app
 
-# Copy your app.py and model files
+# Download models BEFORE copying the rest of the code (to speed up rebuilds)
+# Model 1: Small (360M)
+RUN python3 -c "from huggingface_hub import hf_hub_download; hf_hub_download(repo_id='CyberCoder225/maira-model', filename='SmolLM2-360M-Instruct.Q4_K_M.gguf', local_dir='.')"
+
+# Model 2: Medium (1B)
+RUN python3 -c "from huggingface_hub import hf_hub_download; hf_hub_download(repo_id='bartowski/Llama-3.2-1B-Instruct-GGUF', filename='Llama-3.2-1B-Instruct-Q4_K_M.gguf', local_dir='.')"
+
+# Copy your app.py and brain.py files
 COPY . .
 
 # Start the engine
+EXPOSE 7860
 CMD ["python", "app.py"]