Update entrypoint.sh

entrypoint.sh

@@ -1,43 +1,43 @@
 #!/bin/bash
+set -e
 
-echo "🔥 Axon v6 - llama.cpp Edition"
-echo "📥 Downloading models..."
+# 1. Define the directory
+MODEL_DIR="/data/models"
 
-mkdir -p /data/models
+# 2. Print status
+echo "--- ENTRYPOINT STARTED ---"
+echo "Checking model directory: $MODEL_DIR"
 
-python3 << 'EOF'
-from huggingface_hub import hf_hub_download
+# 3. Create directory if it doesn't exist
+mkdir -p $MODEL_DIR
 
-models = [
-    # Qwen3 30B MoE (Best!)
-    ("unsloth/Qwen3-Coder-30B-A3B-Instruct-GGUF", "Qwen3-Coder-30B-A3B-Instruct-Q4_K_M.gguf"),
-    
-    # Qwen2.5 family
-    ("Qwen/Qwen2.5-Coder-14B-Instruct-GGUF", "qwen2.5-coder-14b-instruct-q4_k_m.gguf"),
-    ("Qwen/Qwen2.5-Coder-7B-Instruct-GGUF", "qwen2.5-coder-7b-instruct-q4_k_m.gguf"),
-    ("Qwen/Qwen2.5-Coder-3B-Instruct-GGUF", "qwen2.5-coder-3b-instruct-q4_k_m.gguf"),
-    ("Qwen/Qwen2.5-Coder-1.5B-Instruct-GGUF", "qwen2.5-coder-1.5b-instruct-q4_k_m.gguf"),
-    ("Qwen/Qwen2.5-Coder-0.5B-Instruct-GGUF", "qwen2.5-coder-0.5b-instruct-q4_k_m.gguf"),
-    
-    # DeepSeek
-    ("bartowski/DeepSeek-Coder-V2-Lite-Instruct-GGUF", "DeepSeek-Coder-V2-Lite-Instruct-Q4_K_M.gguf"),
-    ("TheBloke/deepseek-coder-6.7B-instruct-GGUF", "deepseek-coder-6.7b-instruct.Q4_K_M.gguf"),
-]
+# 4. Check if directory is empty
+if [ -z "$(ls -A $MODEL_DIR)" ]; then
+   echo "⚠️  Folder is empty! Starting download..."
+   
+   # Run the download script (using the Python snippet usually found here)
+   # We use python to leverage the huggingface_hub library for faster/resumable downloads
+   python3 -c "
+from huggingface_hub import snapshot_download
+import os
 
-for repo, filename in models:
-    print(f"📥 Downloading {filename}...")
-    try:
-        hf_hub_download(
-            repo_id=repo,
-            filename=filename,
-            local_dir="/data/models",
-        )
-        print(f"✅ {filename} ready!")
-    except Exception as e:
-        print(f"⚠️ {filename}: {e}")
+print('⬇️  Downloading models to /data/models...')
+try:
+    snapshot_download(
+        repo_id='AIencoder/Axon-Models',
+        local_dir='/data/models',
+        local_dir_use_symlinks=False,
+        repo_type='model'
+    )
+    print('✅ Download complete!')
+except Exception as e:
+    print(f'❌ Download failed: {e}')
+"
+else
+   echo "✅ Models found in $MODEL_DIR. Skipping download."
+   ls -lh $MODEL_DIR
+fi
 
-print("✅ All models downloaded!")
-EOF
-
-echo "🚀 Starting Axon..."
-python3 /app.py
+# 5. Hand over control to the main app
+echo "--- STARTING APP ---"
+exec "$@"