Update api.py

api.py

@@ -112,10 +112,17 @@ async def ipredict(image: UploadFile = File(...)):
 @app.post("/vpredict")
 async def vpredict(image: UploadFile = File(...)):
     if image is not None:
-        img = Image.open(io.BytesIO(await image.read()))
-        # img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
-        img = cv2.imdecode(np.array(img), cv2.IMREAD_COLOR)
-        print("IAM HERE **********************************************************************************")
+        # img = Image.open(io.BytesIO(await image.read()))
+        # # img = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
+        # img = cv2.imdecode(np.array(img), cv2.IMREAD_COLOR)
+        # print("IAM HERE **********************************************************************************")
+        image_bytes = await image.read()
+
+        # Convert bytes to a NumPy uint8 array
+        nparr = np.frombuffer(image_bytes, np.uint8)
+        
+        # Decode into OpenCV image (BGR format)
+        img = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
         # img = img.resize((128, 128))
         # img_array = np.array(img) / 255.0
         # Get the prediction