Ivan Shelonik commited on
Commit
8352ac3
·
1 Parent(s): 29ff838

fix: demo images text excluded

Browse files
Files changed (1) hide show
  1. 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 'Invalid image shape. Expected (28, 28), take from <demo images> folder.'
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)