소스 검색

fix(magic_pdf): add warning for Lite model usage in doc analysis

myhloli 1 년 전
부모
커밋
5be6ee8fce
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  1. 2 0
      magic_pdf/model/doc_analyze_by_custom_model.py

+ 2 - 0
magic_pdf/model/doc_analyze_by_custom_model.py

@@ -68,6 +68,8 @@ def custom_model_init(ocr: bool = False, show_log: bool = False):
     model = None
 
     if model_config.__model_mode__ == "lite":
+        logger.warning("The Lite mode is provided for developers to conduct testing only, and the output quality is "
+                       "not guaranteed to be reliable.")
         model = MODEL.Paddle
     elif model_config.__model_mode__ == "full":
         model = MODEL.PEK