recognize

main.py

@@ -210,7 +210,7 @@ async def analyze_image(image_paths: list[UploadFile]):
                 text_mask = get_not_white_mask(img)
                 masked_img = cv2.bitwise_and(img, img, mask=text_mask)
                 # result = reader.readtext(candidate["img"])
-                result = reader.ocr(masked_img, det=False)
+                result = reader.recognize([masked_img])
                 print(result)
                 if len(result) > 0:
                     ocr_results.append(result)
@@ -232,7 +232,7 @@ async def analyze_image(image_paths: list[UploadFile]):
         # reds = []
         for i, result in enumerate(ocr_results):
             # text = result[0]["generated_text"].upper().strip()
-            text = result[0].get("rec_texts", [])
+            text = result[0][0]
             if text:
                 text = text[0]
             print(text, is_text_valid(text))