add file

main.py

@@ -4,6 +4,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from rapidocr_onnxruntime import RapidOCR
 import io
 import numpy as np
+import pandas as pd
 # det_model_path同理
 model = RapidOCR()
 
@@ -30,8 +31,8 @@ app.add_middleware(
 @app.get("/")
 def read_root():
     return {"Hello": "World!"}
-	
-@app.post("/ocr")	
+    
+@app.post("/ocr")    
 async def ocr(file: UploadFile = File(...)):
     # 读取上传的文件内容
     contents = await file.read()
@@ -39,5 +40,10 @@ async def ocr(file: UploadFile = File(...)):
     image = Image.open(io.BytesIO(contents))
     # 将图像转换为numpy数组
     np_array = np.array(image)
-    result, elapse = model(np_array)
-    return result
+    ocr_result, elapse = model(np_array)
+    dt_boxes, rec_res, scores = list(zip(*ocr_result))
+    out_df = pd.DataFrame(
+        [[box, rec, score] for box, rec, score in zip(dt_boxes, rec_res, scores)],
+        columns=("box", "rec", "score"),
+    )
+    return out_df.to_dict(orient='records')