Create download_models.py

download_models.py

@@ -0,0 +1,35 @@
+import urllib.request
+import sys
+from pathlib import Path
+
+MODELS_DIR = Path("models")
+MODELS_DIR.mkdir(exist_ok=True)
+
+BASE = "https://huggingface.co/Xenova/clip-vit-base-patch32/resolve/main"
+
+FILES = [
+    (f"{BASE}/onnx/vision_model.onnx", MODELS_DIR / "clip_visual.onnx"),
+    (f"{BASE}/onnx/text_model.onnx",   MODELS_DIR / "clip_text.onnx"),
+    (f"{BASE}/tokenizer.json",         MODELS_DIR / "tokenizer.json"),
+]
+
+def progress(block, block_size, total):
+    done = block * block_size
+    if total > 0:
+        pct = min(done / total * 100, 100)
+        sys.stdout.write(f"\r  {pct:.1f}%")
+        sys.stdout.flush()
+
+for url, dest in FILES:
+    if dest.exists():
+        print(f"[=] {dest.name} already exists")
+        continue
+    print(f"[*] Downloading {dest.name}...")
+    try:
+        urllib.request.urlretrieve(url, dest, reporthook=progress)
+        print(f"\n[+] {dest.name} ({dest.stat().st_size // 1024 // 1024}MB)")
+    except Exception as e:
+        print(f"\n[-] Failed: {e}")
+        raise
+
+print("\n[DONE] Models ready")