Update src/services/ocr/replicate_ocr/replicate_ocr.py

src/services/ocr/replicate_ocr/replicate_ocr.py

@@ -4,6 +4,8 @@ created @ 2024-11-06
 author  @ github.com/ishworrsubedii
 """
 import base64
+
+import cv2
 import numpy as np
 import replicate
 
@@ -20,16 +22,15 @@ class ReplicateOCR:
     def __init__(self):
         pass
 
-    def read_text(self, image_path: str):
-        with open(image_path, "rb") as image_file:
-            base_64_image = base64.b64encode(image_file.read()).decode('utf-8')
+    def read_text(self, image_path):
+        _, buffer = cv2.imencode('.png', image_path)
+        base_64_image = base64.b64encode(buffer).decode('utf-8')
 
-        input_image_uri= f"data:image/png;base64,{base_64_image}"
+        input_image_uri = f"data:image/png;base64,{base_64_image}"
         input = {
             "image": input_image_uri,
         }
         output = replicate_run(input)
         output = output.replace("\n", " ").replace("\t", " ")
 
-
-        return output
+        return output