Update download_models.py

download_models.py

@@ -4,35 +4,32 @@ from huggingface_hub import hf_hub_download
 MODEL_DIR = os.path.join(os.path.dirname(__file__), "models")
 os.makedirs(MODEL_DIR, exist_ok=True)
 
-REPO_ID = "monkt/paddleocr-onnx"
-FILES = [
-    "detection/v5/det.onnx",
+# Mobile-детекция (4.83 МБ) вместо server-варианта (88 МБ)
+det_repo = "ilaylow/PP_OCRv5_mobile_onnx"
+det_file = "ppocrv5_det.onnx"
+
+# Рекогнишн под кириллицу + латиницу
+rec_repo = "monkt/paddleocr-onnx"
+REC_FILES = [
     "languages/eslav/rec.onnx",
     "languages/eslav/dict.txt",
 ]
 
-for remote_path in FILES:
+local_det = os.path.join(MODEL_DIR, det_file)
+if not os.path.exists(local_det):
+    print(f"Downloading {det_file}...")
+    hf_hub_download(repo_id=det_repo, filename=det_file, local_dir=MODEL_DIR)
+    print(f"  Saved to {local_det}")
+else:
+    print(f"Skipping {det_file}")
+
+for remote_path in REC_FILES:
     local_path = os.path.join(MODEL_DIR, remote_path)
     if os.path.exists(local_path):
         print(f"Skipping {remote_path}")
         continue
     print(f"Downloading {remote_path}...")
-    hf_hub_download(repo_id=REPO_ID, filename=remote_path, local_dir=MODEL_DIR)
+    hf_hub_download(repo_id=rec_repo, filename=remote_path, local_dir=MODEL_DIR)
     print(f"  Saved to {local_path}")
 
-# Попробуем скачать cls модель
-for repo, path in [
-    ("RapidAI/RapidOCR", "onnx/PP-OCRv4/cls/ch_ppocr_mobile_v2.0_cls_infer.onnx"),
-    ("RapidAI/RapidOCR", "onnx/PP-OCRv3/cls/ch_ppocr_mobile_v2.0_cls_infer.onnx"),
-]:
-    local = os.path.join(MODEL_DIR, path)
-    if os.path.exists(local):
-        continue
-    try:
-        hf_hub_download(repo_id=repo, filename=path, local_dir=MODEL_DIR)
-        print(f"Saved cls: {local}")
-        break
-    except Exception:
-        pass
-
 print("\nAll models ready!")