Fu DaiandClaude Opus 4.8
1a585693be
FastAPI service wrapping the Surya-OCR-2 model (datalab-to) served through vLLM: legacy /v1/api/ai/* endpoints, an OpenAI-compatible /v1/chat/completions endpoint, a coalescing request batcher, a local OCR CLI, Docker packaging, multilingual example outputs, and quantization/concurrency benchmarks. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
24 lines
853 B
Python
24 lines
853 B
Python
from typing import List, Optional
|
|
import os
|
|
import requests
|
|
|
|
from surya.settings import settings
|
|
|
|
|
|
def get_font_path(langs: Optional[List[str]] = None) -> str:
|
|
font_path = settings.RECOGNITION_RENDER_FONTS["all"]
|
|
if langs is not None:
|
|
for k in settings.RECOGNITION_RENDER_FONTS:
|
|
if k in langs and len(langs) == 1:
|
|
font_path = settings.RECOGNITION_RENDER_FONTS[k]
|
|
break
|
|
|
|
if not os.path.exists(font_path):
|
|
os.makedirs(os.path.dirname(font_path), exist_ok=True)
|
|
font_dl_path = f"{settings.RECOGNITION_FONT_DL_BASE}/{os.path.basename(font_path)}"
|
|
with requests.get(font_dl_path, stream=True) as r, open(font_path, 'wb') as f:
|
|
r.raise_for_status()
|
|
for chunk in r.iter_content(chunk_size=8192):
|
|
f.write(chunk)
|
|
|
|
return font_path |