Skip to content

Commit 8f44467

Browse files
author
marwan37
committed
refactor main.py -- remove ground-trith-texts flag
1 parent a33b38c commit 8f44467

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

omni-reader/main.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,10 @@ def get_image_paths(directory: str) -> List[str]:
2929
if __name__ == "__main__":
3030
parser = argparse.ArgumentParser(description="Run OCR comparison pipeline")
3131
parser.add_argument(
32-
"--image_dir", type=str, default="assets", help="Directory containing images to process (default: assets)"
32+
"--image_dir",
33+
type=str,
34+
default="assets",
35+
help="Directory containing images to process (default: assets)",
3336
)
3437
parser.add_argument("--custom_prompt", type=str, help="Custom prompt to use for OCR models")
3538
parser.add_argument(
@@ -59,7 +62,6 @@ def get_image_paths(directory: str) -> List[str]:
5962
pipeline_run = ocr_comparison_pipeline(
6063
images=image_paths,
6164
custom_prompt=args.custom_prompt,
62-
ground_truth_texts=args.ground_truth,
6365
)
6466

6567
print(f"\nPipeline run completed: {pipeline_run.id}")

0 commit comments

Comments
 (0)