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)
|