Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -174,7 +174,7 @@ async def predict(payload: ImagePayload):
|
|
| 174 |
return {"error": "No image provided"}
|
| 175 |
|
| 176 |
if img_base64.startswith("data:image"):
|
| 177 |
-
|
| 178 |
# Preprocess
|
| 179 |
processed_img = preprocess_image(image)
|
| 180 |
|
|
|
|
| 174 |
return {"error": "No image provided"}
|
| 175 |
|
| 176 |
if img_base64.startswith("data:image"):
|
| 177 |
+
img_base64 = img_base64.split(",")[1]
|
| 178 |
# Preprocess
|
| 179 |
processed_img = preprocess_image(image)
|
| 180 |
|