Update main.py

main.py

@@ -92,14 +92,17 @@ def read_img(img):
 app = FastAPI()
 class ImageType(BaseModel):
  url: str
-@app.post("/api/ocr", response_model=dict) 
-def ocr(request: Request, 
- file: bytes = File(...)):
-if request.method == "POST":
- image_stream = io.BytesIO(file)
- 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)
- return {"ImageText" : label}
- return "No post request found"
+@app.post("/api/ocr", response_model=dict)
+async def ocr(request: Request,
+              file: bytes = File(...)
+             ):
+# if request.method == "POST":
+    try:
+        image_stream = io.BytesIO(file)
+        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)
+    except error:
+        return {"Error":"No post request found"}
+    return {"ImageText" : label}