Update main.py

main.py

@@ -173,9 +173,8 @@ async def predict(payload: ImagePayload):
     if not img_base64:
         return {"error": "No image provided"}
 
-    image_data = base64.b64decode(img_base64)
-    image = Image.open(io.BytesIO(image_data))
-
+    if img_base64.startswith("data:image"):
+    img_base64 = img_base64.split(",")[1]
     # Preprocess
     processed_img = preprocess_image(image)