
* Move expensive imports closer to usage Signed-off-by: William Easton <bill.easton@elastic.co> * DCO Remediation Commit for William Easton <bill.easton@elastic.co> I, William Easton <bill.easton@elastic.co>, hereby add my Signed-off-by to this commit: 8a7412ce5bb131a01bb6403067aeb948c9093b0b Signed-off-by: William Easton <bill.easton@elastic.co> * formatting fixes Signed-off-by: William Easton <bill.easton@elastic.co> * DCO Remediation Commit for William Easton <bill.easton@elastic.co> I, William Easton <bill.easton@elastic.co>, hereby add my Signed-off-by to this commit: 8a7412ce5bb131a01bb6403067aeb948c9093b0b I, William Easton <bill.easton@elastic.co>, hereby add my Signed-off-by to this commit: 963e34325071db5e844841f10c27b396a054a0a1 Signed-off-by: William Easton <bill.easton@elastic.co> * Fix baseocrmodel test issue Signed-off-by: William Easton <bill.easton@elastic.co> --------- Signed-off-by: William Easton <bill.easton@elastic.co>
29 lines
886 B
Python
29 lines
886 B
Python
def ocr_engines():
|
|
from docling.models.easyocr_model import EasyOcrModel
|
|
from docling.models.ocr_mac_model import OcrMacModel
|
|
from docling.models.rapid_ocr_model import RapidOcrModel
|
|
from docling.models.tesseract_ocr_cli_model import TesseractOcrCliModel
|
|
from docling.models.tesseract_ocr_model import TesseractOcrModel
|
|
|
|
return {
|
|
"ocr_engines": [
|
|
EasyOcrModel,
|
|
OcrMacModel,
|
|
RapidOcrModel,
|
|
TesseractOcrModel,
|
|
TesseractOcrCliModel,
|
|
]
|
|
}
|
|
|
|
|
|
def picture_description():
|
|
from docling.models.picture_description_api_model import PictureDescriptionApiModel
|
|
from docling.models.picture_description_vlm_model import PictureDescriptionVlmModel
|
|
|
|
return {
|
|
"picture_description": [
|
|
PictureDescriptionVlmModel,
|
|
PictureDescriptionApiModel,
|
|
]
|
|
}
|