Update Dockerfile

Dockerfile

@@ -25,6 +25,22 @@ RUN python3 -c "from huggingface_hub import hf_hub_download; hf_hub_download(rep
 # 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='.')"
 
+# ... (Keep your existing RUN pip and apt-get lines)
+
+# 1. Original Small
+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='.')"
+# 2. Existing Medium (Llama 3.2)
+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='.')"
+
+# --- NEW FEMALE PERSONA MODELS ---
+# 3. Qwen 2.5 1.5B (Very smart/balanced)
+RUN python3 -c "from huggingface_hub import hf_hub_download; hf_hub_download(repo_id='Qwen/Qwen2.5-1.5B-Instruct-GGUF', filename='qwen2.5-1.5b-instruct-q4_k_m.gguf', local_dir='.')"
+# 4. Danube 3 (Fast and chatty)
+RUN python3 -c "from huggingface_hub import hf_hub_download; hf_hub_download(repo_id='h2oai/h2o-danube3-500m-chat-GGUF', filename='h2o-danube3-500m-chat-v1-q4_k_m.gguf', local_dir='.')"
+# 5. StableLM 2 (Elegant/Creative)
+RUN python3 -c "from huggingface_hub import hf_hub_download; hf_hub_download(repo_id='stabilityai/stablelm-2-zephyr-1_6b-gguf', filename='stablelm-2-zephyr-1_6b-Q4_K_M.gguf', local_dir='.')"
+
+# ... (rest of Dockerfile)
 # Copy your app.py and brain.py files
 COPY . .