Mageia commited on
Commit
f99597a
·
unverified ·
1 Parent(s): 7b83985

feat(app): 添加 API 根目录信息

Browse files

- 在 API 根目录 (/docs 或 /redoc) 添加模型名称、设备、用户代理和 OCR 模式信息。
- 允许用户检查 API 状态并获取有关可用 OCR 模式的信息。

Files changed (1) hide show
  1. app.py +14 -1
app.py CHANGED
@@ -108,4 +108,17 @@ async def read_root(request: Request):
108
  """
109
  logger.info(log_message)
110
 
111
- return {"message": "欢迎使用OCR API"}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
108
  """
109
  logger.info(log_message)
110
 
111
+ return {
112
+ "message": "欢迎使用OCR API",
113
+ "user_agent": user_agent,
114
+ "model": model_name,
115
+ "device": device,
116
+ "ocr_mode": [
117
+ "plain texts OCR",
118
+ "format texts OCR",
119
+ "plain multi-crop OCR",
120
+ "format multi-crop OCR",
121
+ "plain fine-grained OCR",
122
+ "format fine-grained OCR",
123
+ ],
124
+ }