Update api.py

api.py

@@ -98,6 +98,7 @@ def predict(image):
 async def ipredict(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 = img.resize((128, 128))
         # img_array = np.array(img) / 255.0
         # Get the prediction
@@ -112,6 +113,7 @@ async def ipredict(image: UploadFile = File(...)):
 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 = img.resize((128, 128))
         # img_array = np.array(img) / 255.0
         # Get the prediction