WangZiFan/DotsOCR
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update app.py

Browse Source
This commit is contained in:
WangZiFan
2026-03-10 16:59:53 +08:00
parent 252bc49307
commit ee38f71f40
1 changed files with 60 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app.py
View File

@@ -41,16 +41,19 @@ async def predict(file: UploadFile = File(...)):
image = Image.open(io.BytesIO(image_bytes)).convert("RGB") image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
except Exception: except Exception:
raise HTTPException(status_code=400, detail="无法识别的图像文件，请上传有效的图像") raise HTTPException(status_code=400, detail="无法识别的图像文件，请上传有效的图像")
logger.info(f"{datetime.now()} - 处理图像: {file.filename}, 尺寸: {image.size}")
logger.info(f"处理图像: {file.filename}, 尺寸: {image.size}")
result = ocr_engine.sample_inference(image) result = ocr_engine.sample_inference(image)
if isinstance(result, dict) and "error" in result:
logger.error(f"{datetime.now()} - 推理返回错误: {result}")
raise HTTPException(status_code=500, detail=result)
return result return result
except HTTPException: except HTTPException:
# 直接抛出 HTTP 异常，避免被通用异常捕获导致状态码错误 # 直接抛出 HTTP 异常，避免被通用异常捕获导致状态码错误
raise raise
except Exception as e: except Exception as e:
logger.exception("推理过程中发生错误") logger.exception(f"{datetime.now()} - 推理过程中发生错误: {e}")
raise HTTPException(status_code=500, detail=str(e)) raise HTTPException(status_code=500, detail=str(e))
@app.get("/health") @app.get("/health")