فهرست منبع

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