reformatted all

Signed-off-by: Peter Staar <taa@zurich.ibm.com>
This commit is contained in:
Peter Staar 2024-09-12 12:39:08 +02:00
parent 3757c61703
commit 998f589495

View File

@ -151,12 +151,15 @@ if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Process PDF files with optional OCR.") parser = argparse.ArgumentParser(description="Process PDF files with optional OCR.")
# Add arguments # Add arguments
parser.add_argument("--pdf", type=str, help="Path to the PDF file.") parser.add_argument("--pdf",
type=str,
default="./tests/data/2206.01062.pdf",
help="Path to the PDF file.")
parser.add_argument( parser.add_argument(
"--ocr", type=bool, default=False, help="Enable OCR (True or False)." "--ocr", type=bool,
default=False,
help="Enable OCR (True or False)."
) )
# Add the backend option as an enum
parser.add_argument( parser.add_argument(
"--backend", "--backend",
type=lambda b: Backend[b.upper()], type=lambda b: Backend[b.upper()],