Update app.py
This commit is contained in:
9
app.py
9
app.py
@@ -41,16 +41,19 @@ async def predict(file: UploadFile = File(...)):
|
||||
image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
|
||||
except Exception:
|
||||
raise HTTPException(status_code=400, detail="无法识别的图像文件,请上传有效的图像")
|
||||
|
||||
logger.info(f"处理图像: {file.filename}, 尺寸: {image.size}")
|
||||
logger.info(f"{datetime.now()} - 处理图像: {file.filename}, 尺寸: {image.size}")
|
||||
|
||||
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
|
||||
except HTTPException:
|
||||
# 直接抛出 HTTP 异常,避免被通用异常捕获导致状态码错误
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.exception("推理过程中发生错误")
|
||||
logger.exception(f"{datetime.now()} - 推理过程中发生错误: {e}")
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
@app.get("/health")
|
||||
|
||||
Reference in New Issue
Block a user