aje6 commited on
Commit
d91f74d
·
verified ·
1 Parent(s): 2ba022b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)