Update main.py

main.py

@@ -27,22 +27,15 @@ async def ocr(
     image: UploadFile = File(...),
     # languages: list = Body(["eng"])
 ):
-    
     try:
         content = await image.read()
         image = Image.open(BytesIO(content))
-        print("[image]",image)
-        if hasattr(pytesseract, "image_to_string"):
-            print("Image to string function is available")
-            print(pytesseract.image_to_string(image, lang = 'eng'))
-            text = ocr_tesseract(image, ['eng'])
-        else:
-            print("Image to string function is not available")
+        text = pytesseract.image_to_string(image, lang = 'eng')
         # text = pytesseract.image_to_string(image, lang="+".join(languages))
     except Exception as e:
         return {"error": str(e)}, 500
 
-    return {"ImageText": "text"}
+    return {"ImageText": text}
 
 @app.post("/api/translate", response_model=dict)
 async def translate(
@@ -51,8 +44,6 @@ async def translate(
     src: str = "en",
     trg: str = "zh",
 ):
-    if api_key != API_KEY:
-        return {"error": "Invalid API key"}, 401
 
     tokenizer, model = get_model(src, trg)