Update main.py

main.py

@@ -41,11 +41,11 @@ async def ocr(
     # languages: list = Body(["eng"])
 ):
     try:
-        print("[Hello]", image.filename.split(".")[-1].lower())
-        # Check if the file format is allowed
-        file_extension = image.filename.split(".")[-1].lower()
-        if file_extension not in VALID_IMAGE_EXTENSIONS:
-            raise HTTPException(status_code=400, detail="Invalid file format. Only .jpg, .jpeg, and .png are allowed.")
+        # print("[Hello]", image.filename.split(".")[-1].lower())
+        # # Check if the file format is allowed
+        # file_extension = image.filename.split(".")[-1].lower()
+        # if file_extension not in VALID_IMAGE_EXTENSIONS:
+        #     raise HTTPException(status_code=400, detail="Invalid file format. Only .jpg, .jpeg, and .png are allowed.")
             
         content = await image.read()
         image = Image.open(BytesIO(content))