Spaces:
Paused
Paused
fix(hf): add hf_transfer for HF_HUB_ENABLE_HF_TRANSFER; remove deprecated symlink arg; add root / route
Browse files- pyproject.toml +1 -0
- requirements.txt +1 -0
- src/kybtech_dots_ocr/app.py +6 -0
- src/kybtech_dots_ocr/model_loader.py +0 -1
pyproject.toml
CHANGED
|
@@ -23,6 +23,7 @@ dependencies = [
|
|
| 23 |
"qwen-vl-utils",
|
| 24 |
"requests>=2.31.0",
|
| 25 |
"httpx>=0.27.0",
|
|
|
|
| 26 |
]
|
| 27 |
|
| 28 |
[project.optional-dependencies]
|
|
|
|
| 23 |
"qwen-vl-utils",
|
| 24 |
"requests>=2.31.0",
|
| 25 |
"httpx>=0.27.0",
|
| 26 |
+
"hf_transfer>=0.1.6",
|
| 27 |
]
|
| 28 |
|
| 29 |
[project.optional-dependencies]
|
requirements.txt
CHANGED
|
@@ -13,3 +13,4 @@ accelerate>=0.20.0
|
|
| 13 |
qwen-vl-utils
|
| 14 |
requests>=2.31.0
|
| 15 |
httpx>=0.27.0
|
|
|
|
|
|
| 13 |
qwen-vl-utils
|
| 14 |
requests>=2.31.0
|
| 15 |
httpx>=0.27.0
|
| 16 |
+
hf_transfer>=0.1.6
|
src/kybtech_dots_ocr/app.py
CHANGED
|
@@ -80,6 +80,12 @@ app = FastAPI(
|
|
| 80 |
)
|
| 81 |
|
| 82 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 83 |
@app.get("/health")
|
| 84 |
async def health_check():
|
| 85 |
"""Health check endpoint."""
|
|
|
|
| 80 |
)
|
| 81 |
|
| 82 |
|
| 83 |
+
@app.get("/")
|
| 84 |
+
async def root():
|
| 85 |
+
"""Root route for uptime checks."""
|
| 86 |
+
return {"status": "ok", "service": "kybtech-dots-ocr", "version": "1.0.0"}
|
| 87 |
+
|
| 88 |
+
|
| 89 |
@app.get("/health")
|
| 90 |
async def health_check():
|
| 91 |
"""Health check endpoint."""
|
src/kybtech_dots_ocr/model_loader.py
CHANGED
|
@@ -83,7 +83,6 @@ class DotsOCRModelLoader:
|
|
| 83 |
local_path = snapshot_download(
|
| 84 |
repo_id=REPO_ID,
|
| 85 |
local_dir=LOCAL_DIR,
|
| 86 |
-
local_dir_use_symlinks=False, # Avoid symlink issues in containers
|
| 87 |
)
|
| 88 |
|
| 89 |
logger.info(f"Model downloaded successfully to {local_path}")
|
|
|
|
| 83 |
local_path = snapshot_download(
|
| 84 |
repo_id=REPO_ID,
|
| 85 |
local_dir=LOCAL_DIR,
|
|
|
|
| 86 |
)
|
| 87 |
|
| 88 |
logger.info(f"Model downloaded successfully to {local_path}")
|