Update download_models.py

download_models.py

@@ -20,4 +20,19 @@ for remote_path in FILES:
     hf_hub_download(repo_id=REPO_ID, 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!")