mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-06-29 15:31:05 +08:00
Fix paddle ocr / minerU cannot add (#15858)
Fix paddle ocr / minerU cannot add
This commit is contained in:
@@ -24,13 +24,13 @@ def normalize_layout_recognizer(layout_recognizer_raw: Any) -> tuple[Any, str |
|
||||
if isinstance(layout_recognizer_raw, str):
|
||||
lowered = layout_recognizer_raw.lower()
|
||||
if lowered.endswith("@mineru"):
|
||||
parser_model_name = layout_recognizer_raw.rsplit("@", 1)[0]
|
||||
parser_model_name = layout_recognizer_raw
|
||||
layout_recognizer = "MinerU"
|
||||
elif lowered.endswith("@paddleocr"):
|
||||
parser_model_name = layout_recognizer_raw.rsplit("@", 1)[0]
|
||||
parser_model_name = layout_recognizer_raw
|
||||
layout_recognizer = "PaddleOCR"
|
||||
elif lowered.endswith("@opendataloader"):
|
||||
parser_model_name = layout_recognizer_raw.rsplit("@", 1)[0]
|
||||
parser_model_name = layout_recognizer_raw
|
||||
layout_recognizer = "OpenDataLoader"
|
||||
|
||||
return layout_recognizer, parser_model_name
|
||||
|
||||
Reference in New Issue
Block a user