Skip to content

Commit

Permalink
Update detector checkpoint
Browse files Browse the repository at this point in the history
  • Loading branch information
VikParuchuri committed Feb 6, 2025
1 parent 2286b4c commit 4cfc55d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion surya/settings.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 4cfc55d

Please sign in to comment.