ソースを参照

fix(scripts): update model download scripts for OCR

- Update download_models.py and download_models_hf.py scripts
- Change OCR model path from paddleocr to paddleocr_torch
myhloli 7 ヶ月 前
コミット
5237a3851c
2 ファイル変更2 行追加2 行削除
  1. 1 1
      scripts/download_models.py
  2. 1 1
      scripts/download_models_hf.py

+ 1 - 1
scripts/download_models.py

@@ -37,7 +37,7 @@ if __name__ == '__main__':
         "models/Layout/YOLO/*",
         "models/MFD/YOLO/*",
         "models/MFR/unimernet_hf_small_2503/*",
-        "models/OCR/paddleocr/*",
+        "models/OCR/paddleocr_torch/*",
         # "models/TabRec/TableMaster/*",
         # "models/TabRec/StructEqTable/*",
     ]

+ 1 - 1
scripts/download_models_hf.py

@@ -38,7 +38,7 @@ if __name__ == '__main__':
         "models/Layout/YOLO/*",
         "models/MFD/YOLO/*",
         "models/MFR/unimernet_hf_small_2503/*",
-        "models/OCR/paddleocr/*",
+        "models/OCR/paddleocr_torch/*",
         # "models/TabRec/TableMaster/*",
         # "models/TabRec/StructEqTable/*",
     ]