Update start.sh

start.sh

@@ -1,10 +1,18 @@
 #!/bin/bash
 set -e
 
+echo "================================================"
+echo "🚀 Axon - Qwen2.5-Omni-7B Multimodal Server"
+echo "================================================"
+echo ""
+echo "📋 Capabilities: Text | Images | Audio"
+echo "🔧 Quantization: Q8_0 (near-lossless)"
+echo ""
 echo "------------------------------------------------"
 echo "🛠️  Setting up Environment"
 echo "------------------------------------------------"
 
+# Create model directory structure
 mkdir -p /app/models/qwen2.5-omni-7b
 
 download_file () {
@@ -12,28 +20,51 @@ download_file () {
     FILENAME=$2
     DEST_PATH=$3
 
-    if [ ! -f "$DEST_PATH" ]; then
-        echo "⬇️  Downloading $FILENAME ..."
-        python3 -c "
+    if [ -f "$DEST_PATH" ]; then
+        echo "✅ Already exists: $(basename $DEST_PATH)"
+        return 0
+    fi
+
+    echo "⬇️  Downloading $FILENAME ..."
+    python3 -c "
 from huggingface_hub import hf_hub_download
-import shutil, sys
+import shutil
+import sys
+
 try:
-    path = hf_hub_download(repo_id='$REPO_ID', filename='$FILENAME')
+    path = hf_hub_download(
+        repo_id='$REPO_ID',
+        filename='$FILENAME',
+        cache_dir='/app/.cache'
+    )
     shutil.copy(path, '$DEST_PATH')
+    print(f'✅ Downloaded: $FILENAME')
 except Exception as e:
-    print(f'Error downloading $FILENAME: {e}')
+    print(f'❌ Error downloading $FILENAME: {e}')
     sys.exit(1)
 "
-    fi
 }
 
-# Qwen2.5-Omni-7B Q8_0 - Text + Images + Audio (near-lossless quality)
-download_file "ggml-org/Qwen2.5-Omni-7B-GGUF" "Qwen2.5-Omni-7B-Q8_0.gguf" "/app/models/qwen2.5-omni-7b/Qwen2.5-Omni-7B-Q8_0.gguf"
-download_file "ggml-org/Qwen2.5-Omni-7B-GGUF" "mmproj-Qwen2.5-Omni-7B-Q8_0.gguf" "/app/models/qwen2.5-omni-7b/mmproj-Qwen2.5-Omni-7B-Q8_0.gguf"
+echo ""
+echo "📦 Downloading Qwen2.5-Omni-7B Q8_0..."
+echo ""
+
+# Download model and multimodal projector
+download_file "ggml-org/Qwen2.5-Omni-7B-GGUF" \
+    "Qwen2.5-Omni-7B-Q8_0.gguf" \
+    "/app/models/qwen2.5-omni-7b/Qwen2.5-Omni-7B-Q8_0.gguf"
+
+download_file "ggml-org/Qwen2.5-Omni-7B-GGUF" \
+    "mmproj-Qwen2.5-Omni-7B-Q8_0.gguf" \
+    "/app/models/qwen2.5-omni-7b/mmproj-Qwen2.5-Omni-7B-Q8_0.gguf"
 
+echo ""
 echo "------------------------------------------------"
-echo "🚀 Starting Server"
+echo "🚀 Starting llama.cpp Server"
 echo "------------------------------------------------"
+echo ""
+echo "🌐 Server will be available at http://0.0.0.0:7860"
+echo ""
 
 exec /usr/local/bin/llama-server \
     --models-dir /app/models \