Update api.py

api.py

@@ -9,6 +9,7 @@ import numpy as np
 import urllib.request
 import io
 import os
+from ofa_ocr import ofa_ocr_gr ##导入ofa_ocr中的大函数,方便使用其获取坐标的子函数
 
 app = FastAPI()
 
@@ -25,13 +26,6 @@ async def getCoordinates(image: UploadFile = File(...)):
     image = Image.open(io.BytesIO(img)).convert("L").convert("RGB")
     path_to_image = "save.jpg"
     image.save(path_to_image)
-    result = predict_detections_and_associations(
-            path_to_image,
-            input_character_detection_threshold,
-            input_panel_detection_threshold,
-            input_text_detection_threshold,
-            input_character_character_matching_threshold,
-            input_text_character_matching_threshold,
-        )
+    result = ofa_ocr_gr.ocr_api(path_to_image)[2]
     os.remove(path_to_image)
     return result