handsme commited on
Commit
a469cb9
·
1 Parent(s): 57d81fa

修复 elapsed 处理

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -53,12 +53,15 @@ def ocr_recognize(image):
53
  "bbox": bbox if isinstance(bbox, list) else bbox.tolist(),
54
  })
55
 
 
 
 
56
  return json.dumps({
57
  "success": True,
58
  "lines": lines,
59
  "full_text": "\n".join(lines),
60
  "raw": raw_results,
61
- "elapsed": round(elapsed, 3),
62
  }, ensure_ascii=False, indent=2)
63
  except Exception as e:
64
  return json.dumps({"success": False, "error": str(e)}, ensure_ascii=False)
 
53
  "bbox": bbox if isinstance(bbox, list) else bbox.tolist(),
54
  })
55
 
56
+ # elapsed 可能是包含多个阶段耗时的列表,求和即可
57
+ total_elapsed = sum(elapsed) if isinstance(elapsed, list) else elapsed
58
+
59
  return json.dumps({
60
  "success": True,
61
  "lines": lines,
62
  "full_text": "\n".join(lines),
63
  "raw": raw_results,
64
+ "elapsed": round(total_elapsed, 3),
65
  }, ensure_ascii=False, indent=2)
66
  except Exception as e:
67
  return json.dumps({"success": False, "error": str(e)}, ensure_ascii=False)