Refactor prediction function: update logging to print composite image shape
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ model = tf.keras.models.load_model("./sketch_recognition_numbers_model.h5")
|
|
| 28 |
|
| 29 |
# Prediction function for sketch recognition
|
| 30 |
def predict(data):
|
| 31 |
-
print(data.shape)
|
| 32 |
# Reshape image to 28x28
|
| 33 |
img = np.reshape(data['composite'], (1, img_size, img_size, 1))
|
| 34 |
# Make prediction
|
|
|
|
| 28 |
|
| 29 |
# Prediction function for sketch recognition
|
| 30 |
def predict(data):
|
| 31 |
+
print(data['composite'].shape)
|
| 32 |
# Reshape image to 28x28
|
| 33 |
img = np.reshape(data['composite'], (1, img_size, img_size, 1))
|
| 34 |
# Make prediction
|