Spaces:
Runtime error
Runtime error
Ivan Shelonik
commited on
Commit
·
8352ac3
1
Parent(s):
29ff838
fix: demo images text excluded
Browse files- api_server.py +1 -1
api_server.py
CHANGED
|
@@ -88,7 +88,7 @@ def predict():
|
|
| 88 |
|
| 89 |
# Check image shape
|
| 90 |
if image_data.size != (28, 28):
|
| 91 |
-
return
|
| 92 |
|
| 93 |
# Preprocess the image
|
| 94 |
processed_image = preprocess_image(image_data)
|
|
|
|
| 88 |
|
| 89 |
# Check image shape
|
| 90 |
if image_data.size != (28, 28):
|
| 91 |
+
return "Invalid image shape. Expected (28, 28), take from 'demo images' folder."
|
| 92 |
|
| 93 |
# Preprocess the image
|
| 94 |
processed_image = preprocess_image(image_data)
|