Update download_model.py

download_model.py

@@ -2,14 +2,31 @@ from huggingface_hub import hf_hub_download
 import shutil
 import os
 
-print("Downloading ONNX model from toandev/OCR-for-Captcha...")
 os.makedirs("./model", exist_ok=True)
 
-model_path = hf_hub_download(
-    repo_id="toandev/OCR-for-Captcha",
-    filename="model.onnx",
-    repo_type="space"
-)
+# Try model repo first, then space repo
+configs = [
+    {"repo_id": "toandev/OCR-for-Captcha", "repo_type": "model"},
+    {"repo_id": "toandev/OCR-for-Captcha", "repo_type": "space"},
+    {"repo_id": "toandev/OCR-for-Captcha", "repo_type": "dataset"},
+]
 
-shutil.copy(model_path, "./model/model.onnx")
-print("✅ ONNX model saved to ./model/model.onnx")
+model_path = None
+for cfg in configs:
+    try:
+        print(f"Trying repo_type='{cfg['repo_type']}'...")
+        model_path = hf_hub_download(
+            repo_id=cfg["repo_id"],
+            filename="model.onnx",
+            repo_type=cfg["repo_type"]
+        )
+        print(f"✅ Found with repo_type='{cfg['repo_type']}'")
+        break
+    except Exception as e:
+        print(f"❌ Failed: {e}")
+
+if model_path:
+    shutil.copy(model_path, "./model/model.onnx")
+    print("✅ ONNX model saved to ./model/model.onnx")
+else:
+    raise RuntimeError("Could not find model.onnx in any repo type")