mirror of
https://github.com/infiniflow/ragflow.git
synced 2025-12-23 06:46:40 +08:00
Refa: treat MinerU as an OCR model (#11849)
### What problem does this PR solve? Treat MinerU as an OCR model. ### Type of change - [x] New Feature (non-breaking change which adds functionality) - [x] Refactoring
This commit is contained in:
@ -121,6 +121,7 @@ EmbeddingModel = globals().get("EmbeddingModel", {})
|
||||
RerankModel = globals().get("RerankModel", {})
|
||||
Seq2txtModel = globals().get("Seq2txtModel", {})
|
||||
TTSModel = globals().get("TTSModel", {})
|
||||
OcrModel = globals().get("OcrModel", {})
|
||||
|
||||
|
||||
MODULE_MAPPING = {
|
||||
@ -130,6 +131,7 @@ MODULE_MAPPING = {
|
||||
"rerank_model": RerankModel,
|
||||
"sequence2txt_model": Seq2txtModel,
|
||||
"tts_model": TTSModel,
|
||||
"ocr_model": OcrModel,
|
||||
}
|
||||
|
||||
package_name = __name__
|
||||
@ -171,4 +173,5 @@ __all__ = [
|
||||
"RerankModel",
|
||||
"Seq2txtModel",
|
||||
"TTSModel",
|
||||
"OcrModel",
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user