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

feat add glm-ocr core

Browse Source
This commit is contained in:
liuyuanchuang
2026-03-09 16:51:06 +08:00
parent d74130914c
commit 6dfaf9668b
17 changed files with 1687 additions and 140 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
app/core/config.py
View File

@@ -3,9 +3,8 @@
from functools import lru_cache
from pathlib import Path
from pydantic_settings import BaseSettings, SettingsConfigDict
import torch
from typing import Optional
from pydantic_settings import BaseSettings, SettingsConfigDict
class Settings(BaseSettings):
@@ -48,21 +47,25 @@ class Settings(BaseSettings):
is_padding: bool = True
padding_ratio: float = 0.1
max_tokens: int = 4096
# Model Paths
pp_doclayout_model_dir: Optional[str] = "/home/yoge/.cache/modelscope/hub/models/PaddlePaddle/PP-DocLayoutV3"
pp_doclayout_model_dir: str | None = (
"/home/yoge/.cache/modelscope/hub/models/PaddlePaddle/PP-DocLayoutV3"
)
# Image Processing
max_image_size_mb: int = 10
image_padding_ratio: float = 0.1 # 10% on each side = 20% total expansion
device: torch.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") # cuda:0 or cpu
device: torch.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
# Server Settings
host: str = "0.0.0.0"
port: int = 8053
# Logging Settings
log_dir: Optional[str] = None # Defaults to /app/logs in container or ./logs locally
log_dir: str | None = None # Defaults to /app/logs in container or ./logs locally
log_level: str = "INFO" # DEBUG, INFO, WARNING, ERROR, CRITICAL
@property