Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -250,6 +250,7 @@ def predict(image):
|
|
250 |
image = cv2.resize(image, (input_shape[2], input_shape[3]))
|
251 |
|
252 |
original_image_shape = image.shape
|
|
|
253 |
|
254 |
# Reshape the image to match the model's input shape
|
255 |
image = image.reshape(3, 640, 640)
|
|
|
250 |
image = cv2.resize(image, (input_shape[2], input_shape[3]))
|
251 |
|
252 |
original_image_shape = image.shape
|
253 |
+
print("Original image shape:", original_image_shape)
|
254 |
|
255 |
# Reshape the image to match the model's input shape
|
256 |
image = image.reshape(3, 640, 640)
|