Update app.py

app.py

@@ -692,5 +692,28 @@ with gr.Blocks(title="Axon v6") as demo:
     mock_btn.click(generate_mock_data, [mock_schema, mock_count, mock_format, model_dropdown, max_tokens], mock_output)
     format_btn.click(convert_data_format, [format_input, format_from, format_to, model_dropdown, max_tokens], format_output)
 
+# Pre-download ALL models on startup
 print("🔥 Axon v6 starting...")
+print("⬇️ Pre-downloading all models...")
+
+for model_name in MODELS.keys():
+    print(f"📥 Checking {model_name}...")
+    filename = MODELS[model_name]
+    model_path = os.path.join(MODELS_DIR, filename)
+    
+    if not os.path.exists(model_path):
+        repo_id = MODEL_REPOS.get(filename)
+        if repo_id:
+            print(f"⬇️ Downloading {filename}...")
+            try:
+                hf_hub_download(repo_id=repo_id, filename=filename, local_dir=MODELS_DIR)
+                print(f"✅ {filename} downloaded!")
+            except Exception as e:
+                print(f"❌ Failed to download {filename}: {e}")
+    else:
+        print(f"✅ {filename} already cached")
+
+print("✅ All models ready!")
+print("🚀 Launching Axon...")
+
 demo.launch(server_name="0.0.0.0", server_port=7860, theme=gr.themes.Soft(primary_hue="indigo"))