Browse Source

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

myhloli 1 year ago
parent
commit
5be6ee8fce
1 changed files with 2 additions and 0 deletions
  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