Commit 5be6ee8f authored by myhloli's avatar myhloli

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

parent 2c09109e
...@@ -68,6 +68,8 @@ def custom_model_init(ocr: bool = False, show_log: bool = False): ...@@ -68,6 +68,8 @@ def custom_model_init(ocr: bool = False, show_log: bool = False):
model = None model = None
if model_config.__model_mode__ == "lite": 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 model = MODEL.Paddle
elif model_config.__model_mode__ == "full": elif model_config.__model_mode__ == "full":
model = MODEL.PEK model = MODEL.PEK
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment