Преглед изворни кода

refactor: comment out paddleocr model copying code

- Commented out the code that copies the paddleocr model to user directory
- This change affects both download_models.py and download_models_hf.py scripts
myhloli пре 7 месеци
родитељ
комит
3bd1e0e46f
2 измењених фајлова са 10 додато и 10 уклоњено
  1. 5 5
      scripts/download_models.py
  2. 5 5
      scripts/download_models_hf.py

+ 5 - 5
scripts/download_models.py

@@ -47,11 +47,11 @@ if __name__ == '__main__':
     print(f'model_dir is: {model_dir}')
     print(f'layoutreader_model_dir is: {layoutreader_model_dir}')
 
-    paddleocr_model_dir = model_dir + '/OCR/paddleocr'
-    user_paddleocr_dir = os.path.expanduser('~/.paddleocr')
-    if os.path.exists(user_paddleocr_dir):
-        shutil.rmtree(user_paddleocr_dir)
-    shutil.copytree(paddleocr_model_dir, user_paddleocr_dir)
+    # paddleocr_model_dir = model_dir + '/OCR/paddleocr'
+    # user_paddleocr_dir = os.path.expanduser('~/.paddleocr')
+    # if os.path.exists(user_paddleocr_dir):
+    #     shutil.rmtree(user_paddleocr_dir)
+    # shutil.copytree(paddleocr_model_dir, user_paddleocr_dir)
 
     json_url = 'https://gcore.jsdelivr.net/gh/opendatalab/MinerU@master/magic-pdf.template.json'
     config_file_name = 'magic-pdf.json'

+ 5 - 5
scripts/download_models_hf.py

@@ -54,11 +54,11 @@ if __name__ == '__main__':
     print(f'model_dir is: {model_dir}')
     print(f'layoutreader_model_dir is: {layoutreader_model_dir}')
 
-    paddleocr_model_dir = model_dir + '/OCR/paddleocr'
-    user_paddleocr_dir = os.path.expanduser('~/.paddleocr')
-    if os.path.exists(user_paddleocr_dir):
-        shutil.rmtree(user_paddleocr_dir)
-    shutil.copytree(paddleocr_model_dir, user_paddleocr_dir)
+    # paddleocr_model_dir = model_dir + '/OCR/paddleocr'
+    # user_paddleocr_dir = os.path.expanduser('~/.paddleocr')
+    # if os.path.exists(user_paddleocr_dir):
+    #     shutil.rmtree(user_paddleocr_dir)
+    # shutil.copytree(paddleocr_model_dir, user_paddleocr_dir)
 
     json_url = 'https://github.com/opendatalab/MinerU/raw/master/magic-pdf.template.json'
     config_file_name = 'magic-pdf.json'