Update app.py

app.py

@@ -164,7 +164,12 @@ def predict(upload_image):
     # Load the image
 
     # img = cv2.imread(test_image_path)
-    img = upload_image
+    # img = upload_image
+    # img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    pil_image = upload_image.convert('RGB')
+    open_cv_image = np.array(pil_image)
+    # Convert RGB to BGR
+    img = open_cv_image[:, :, ::-1].copy()
     img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
 
     # Resize the image to 224x224 pixels