Update main.py

main.py

@@ -90,9 +90,9 @@ def load_image_url(source):
         img = np.asarray(bytearray(response.content), dtype=np.uint8)
         img = cv2.imdecode(img, cv2.IMREAD_GRAYSCALE)
     else:
-        img = base64.b64decode(source)
-        img = Image.open(BytesIO(img))
-        img = np.array(img)
+        img_data = base64.b64decode(source)
+        image = Image.open(BytesIO(img_data)).convert("RGB")
+        img = np.array(image).astype(np.uint8) 
         img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
 
     return img