Update main.py

main.py

@@ -5,7 +5,9 @@ from fastapi.security.api_key import APIKeyHeader
 from typing import Optional, Annotated
 from fastapi.encoders import jsonable_encoder
 from PIL import Image
-from io import BytesIO
+import io
+import cv2
+import numpy as np
 import pytesseract
 from nltk.tokenize import sent_tokenize
 from transformers import MarianMTModel, MarianTokenizer
@@ -30,9 +32,14 @@ async def ocr(
         print("[1]",os.popen(f'cat /etc/debian_version').read())
         print("[2]",os.popen(f'cat /etc/issue').read())
         print("[3]",os.popen(f'apt search tesseract').read())
-        content = await image.read()
-        image = Image.open(BytesIO(content))
-        print("[image]",content)
+        # content = await image.read()
+        # image = Image.open(BytesIO(content))
+        image_stream = io.BytesIO(image)
+        image_stream.seek(0)
+        file_bytes = np.asarray(bytearray(image_stream.read()), dtype=np.uint8)
+        frame = cv2.imdecode(file_bytes, cv2.IMREAD_COLOR)
+        # label = read_img(frame)
+        print("[image]",frame)
         if hasattr(pytesseract, "image_to_string"):
             print("Image to string function is available")
         else: