Update installer to download DreamShaper 8 model and sync tool checkpoint name

install.sh

@@ -79,6 +79,20 @@ if [[ $REPLY =~ ^[Yy]$ ]]; then
         pip3 install -r requirements.txt
         echo "✅ ComfyUI installed."
         
+        # 5. Download Image Generation Model
+        echo ""
+        echo "[*] Downloading Image Generation Model (DreamShaper 8)..."
+        MODEL_DIR="models/checkpoints"
+        mkdir -p "$MODEL_DIR"
+        MODEL_PATH="$MODEL_DIR/dreamshaper_8.safetensors"
+        if [ ! -f "$MODEL_PATH" ]; then
+            echo "    This may take a few minutes (approx. 2GB)..."
+            curl -L "https://huggingface.co/Lykon/DreamShaper/resolve/main/DreamShaper_8_pruned.safetensors" -o "$MODEL_PATH"
+            echo "✅ Model downloaded."
+        else
+            echo "✅ Model already exists."
+        fi
+
         echo ""
         echo "[!] To use Image Generation, you must start ComfyUI:"
         echo "    cd $INSTALL_DIR"

src/ollama_cli/tools/media.py

@@ -30,7 +30,7 @@ def generate_image(prompt: str) -> str:
         # Based on the user's previous generate_image.py, but using the configured URL
         workflow = {
             "1": {
-                "inputs": {"ckpt_name": "dreamshaper_8.safetensors"},
+                "inputs": {"ckpt_name": "dreamshaper_8_pruned.safetensors"},
                 "class_type": "CheckpointLoaderSimple"
             },
             "2": {