浏览代码

refactor(magic_pdf): remove unused imports and update dependencies

- Remove unused imports for concurrent.futures, multiprocessing, and paddle
- Delete commented-out code
- Update numpy dependency to remove upper version limit
- Remove InferenceResult import that was commented out
myhloli 7 月之前
父节点
当前提交
243bc58cc6
共有 2 个文件被更改,包括 2 次插入8 次删除
  1. 1 7
      magic_pdf/model/doc_analyze_by_custom_model.py
  2. 1 1
      requirements.txt

+ 1 - 7
magic_pdf/model/doc_analyze_by_custom_model.py

@@ -1,5 +1,3 @@
-import concurrent.futures as fut
-import multiprocessing as mp
 import os
 import os
 import time
 import time
 
 
@@ -25,8 +23,6 @@ from magic_pdf.libs.config_reader import (get_device, get_formula_config,
                                           get_table_recog_config)
                                           get_table_recog_config)
 from magic_pdf.model.model_list import MODEL
 from magic_pdf.model.model_list import MODEL
 
 
-# from magic_pdf.operators.models import InferenceResult
-
 class ModelSingleton:
 class ModelSingleton:
     _instance = None
     _instance = None
     _models = {}
     _models = {}
@@ -244,9 +240,7 @@ def may_batch_image_analyze(
         formula_enable=None,
         formula_enable=None,
         table_enable=None):
         table_enable=None):
     # os.environ['CUDA_VISIBLE_DEVICES'] = str(idx)
     # os.environ['CUDA_VISIBLE_DEVICES'] = str(idx)
-    # 关闭paddle的信号处理
-    import paddle
-    paddle.disable_signal_handler()
+
     from magic_pdf.model.batch_analyze import BatchAnalyze
     from magic_pdf.model.batch_analyze import BatchAnalyze
 
 
     model_manager = ModelSingleton()
     model_manager = ModelSingleton()

+ 1 - 1
requirements.txt

@@ -3,7 +3,7 @@ Brotli>=1.1.0
 click>=8.1.7
 click>=8.1.7
 fast-langdetect>=0.2.3,<0.3.0
 fast-langdetect>=0.2.3,<0.3.0
 loguru>=0.6.0
 loguru>=0.6.0
-numpy>=1.21.6,<2.0.0
+numpy>=1.21.6
 pydantic>=2.7.2
 pydantic>=2.7.2
 PyMuPDF>=1.24.9,<=1.24.14
 PyMuPDF>=1.24.9,<=1.24.14
 scikit-learn>=1.0.2
 scikit-learn>=1.0.2