File size: 722 Bytes
ba8abef | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | #!/usr/bin/env python3
"""OCR an image via the local PaddleOCR-VL server."""
from __future__ import annotations
import argparse
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parent))
from ocr_engine import DEFAULT_BASE_URL, DEFAULT_MODEL, OcrEngine
def main() -> int:
p = argparse.ArgumentParser()
p.add_argument("image", type=Path)
p.add_argument("--base-url", default=DEFAULT_BASE_URL)
p.add_argument("--model", default=DEFAULT_MODEL)
args = p.parse_args()
text, dt, _ = OcrEngine(args.base_url, args.model).ocr_image(args.image)
print(text)
print(f"\n# {dt:.2f}s", file=sys.stderr)
return 0
if __name__ == "__main__":
sys.exit(main())
|