From 4cfc55d3931a8df53836f55cf67b20f2611aabbe Mon Sep 17 00:00:00 2001 From: Vik Paruchuri Date: Thu, 6 Feb 2025 14:34:12 -0500 Subject: [PATCH] Update detector checkpoint --- surya/settings.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/surya/settings.py b/surya/settings.py index 0630445..3a9d257 100644 --- a/surya/settings.py +++ b/surya/settings.py @@ -45,7 +45,7 @@ def TORCH_DEVICE_MODEL(self) -> str: # Text detection DETECTOR_BATCH_SIZE: Optional[int] = None # Defaults to 2 for CPU/MPS, 32 otherwise - DETECTOR_MODEL_CHECKPOINT: str = "datalab-to/line_detector0@885c16e02c4d89113f904991ed88c48c461d0724" + DETECTOR_MODEL_CHECKPOINT: str = "datalab-to/line_detector0@5b7f57b471c54026adb9726ab527c9fbbedac1b7" DETECTOR_BENCH_DATASET_NAME: str = "vikp/doclaynet_bench" DETECTOR_IMAGE_CHUNK_HEIGHT: int = 1400 # Height at which to slice images vertically DETECTOR_TEXT_THRESHOLD: float = 0.6 # Threshold for text detection (above this is considered text)