Review fixes

Signed-off-by: Christoph Auer <cau@zurich.ibm.com>
This commit is contained in:
Christoph Auer 2024-12-13 15:44:06 +01:00
parent d2f114e5d6
commit 6832c5aeba
2 changed files with 2 additions and 2 deletions

View File

@ -135,7 +135,7 @@ class EasyOcrOptions(OcrOptions):
kind: Literal["easyocr"] = "easyocr" kind: Literal["easyocr"] = "easyocr"
lang: List[str] = ["fr", "de", "es", "en"] lang: List[str] = ["fr", "de", "es", "en"]
use_gpu: Annotated[ use_gpu: Annotated[
int, bool,
Field( Field(
deprecated="Deprecated field. Better to set the `accelerator_options.device` in `pipeline_options`. " deprecated="Deprecated field. Better to set the `accelerator_options.device` in `pipeline_options`. "
"When `use_gpu and accelerator_options.device == AcceleratorDevice.CUDA` the GPU is used " "When `use_gpu and accelerator_options.device == AcceleratorDevice.CUDA` the GPU is used "

View File

@ -74,7 +74,7 @@ def main():
pipeline_options.do_ocr = True pipeline_options.do_ocr = True
pipeline_options.do_table_structure = True pipeline_options.do_table_structure = True
pipeline_options.table_structure_options.do_cell_matching = True pipeline_options.table_structure_options.do_cell_matching = True
pipeline_options.ocr_options.lang = "es" pipeline_options.ocr_options.lang = ["es"]
pipeline_options.accelerator_options = AcceleratorOptions( pipeline_options.accelerator_options = AcceleratorOptions(
num_threads=4, device=Device.AUTO num_threads=4, device=Device.AUTO
) )