handsme commited on
Commit
689f815
·
1 Parent(s): 9595b9e

修复:API 端点添加图片缩放防止大图崩溃

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -89,6 +89,13 @@ async def api_predict(request: Request):
89
  image_bytes = base64.b64decode(base64_str)
90
  image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
91
 
 
 
 
 
 
 
 
92
  result_json_str = ocr_recognize(image)
93
  return {"data": [result_json_str]}
94
  except Exception as e:
 
89
  image_bytes = base64.b64decode(base64_str)
90
  image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
91
 
92
+ # 限制图片最大边为 2000px,防止大图导致 PaddlePaddle C++ 层内存溢出
93
+ max_side = 2000
94
+ w, h = image.size
95
+ if max(w, h) > max_side:
96
+ scale = max_side / max(w, h)
97
+ image = image.resize((int(w * scale), int(h * scale)), Image.LANCZOS)
98
+
99
  result_json_str = ocr_recognize(image)
100
  return {"data": [result_json_str]}
101
  except Exception as e: