YogeLiu/doc_processer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: get setting param

Browse Source
This commit is contained in:
liuyuanchuang
2026-02-07 09:11:43 +08:00
parent f0ad0a4c77
commit 23e2160668
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app/core/config.py
View File

@@ -30,8 +30,12 @@ class Settings(BaseSettings):
# GLM OCR Settings # GLM OCR Settings
glm_ocr_url: str = "http://127.0.0.1:8002/v1" glm_ocr_url: str = "http://127.0.0.1:8002/v1"
# padding ratio
is_padding: bool = False
padding_ratio: float = 0.15
# Model Paths # Model Paths
pp_doclayout_model_dir: Optional[str] = "/home/yoge/.cache/modelscope/hub/models/PaddlePaddle/PP-DocLayoutV2" pp_doclayout_model_dir: Optional[str] = "/home/yoge/.cache/modelscope/hub/models/PaddlePaddle/PP-DocLayoutV3"
# Image Processing # Image Processing
max_image_size_mb: int = 10 max_image_size_mb: int = 10

2
app/services/ocr_service.py
View File

@@ -660,7 +660,7 @@ class MineruOCRService(OCRServiceBase):
Dict with 'markdown', 'latex', 'mathml' keys. Dict with 'markdown', 'latex', 'mathml' keys.
""" """
try: try:
if self.image_processor and get_settings().is_padding: if self.image_processor and settings.is_padding:
image = self.image_processor.add_padding(image) image = self.image_processor.add_padding(image)
# Convert numpy array to image bytes # Convert numpy array to image bytes