dots.ocr release

tools/download_model.py

@@ -0,0 +1,19 @@
+from argparse import ArgumentParser
+import os
+
+
+if __name__ == '__main__':
+    parser = ArgumentParser()
+    parser.add_argument('--type', '-t', type=str, default="huggingface")
+    parser.add_argument('--name', '-n', type=str, default="rednote-hilab/dots.ocr")
+    args = parser.parse_args()
+    script_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+    print(f"Attention: The model save dir dots.ocr should be replace by a name without `.` like DotsOCR, util we merge our code to transformers.")
+    model_dir = os.path.join(script_dir, "weights/DotsOCR")
+    if not os.path.exists(model_dir):
+        os.makedirs(model_dir)
+    if args.type == "huggingface":
+        from huggingface_hub import snapshot_download
+        snapshot_download(repo_id=args.name, local_dir=model_dir, local_dir_use_symlinks=False, resume_download=True)
+    
+    print(f"model downloaded to {model_dir}")