Update main.py

main.py

@@ -132,7 +132,6 @@ def valuate_qna(p: str, r: str) -> str:
     return hug.output.get("data", [None])[0]
 
 def ocr(image: io.BytesIO) -> str:
-    image = compress_image(image) # Compress the image
     return process_image(image, os.getenv('AZKEY'), os.getenv('AZURL'))
 
 app = Flask(__name__)
@@ -187,6 +186,9 @@ def task_ocr():
     headers = dict(request.headers)
     if not headers.get("KEY") != os.getenv("KEY"):
         return jsonify({"status" : False, "msg" : "Invalid API Key"}), 404
+    size = len(file_content.getvalue())
+    if size > 4194304:
+        return jsonify({"status" : False, "msg" : "File size is greater then 4MB"}), 404
     answer = ocr(file_content)
     return jsonify({"status" : True, "data" : answer})