rbarman commited on
Commit
69d9453
·
1 Parent(s): 8239d4b
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,10 +22,10 @@ def predict(img: np.ndarray) -> str:
22
  # The MobileNet model expects images in RGB format.
23
  # Resize to MobileNet image shape.
24
  input_image = cv2.resize(src=img, dsize=(224, 224))
25
- print(f'resized: {img.shape}')
26
  # Reshape to model input shape.
27
  input_image = np.expand_dims(input_image, 0)
28
- print(f'final shape: {img.shape}')
29
 
30
  # Get inference result
31
  result_infer = compiled_model([input_image])[output_layer]
 
22
  # The MobileNet model expects images in RGB format.
23
  # Resize to MobileNet image shape.
24
  input_image = cv2.resize(src=img, dsize=(224, 224))
25
+ print(f'resized: {input_image.shape}')
26
  # Reshape to model input shape.
27
  input_image = np.expand_dims(input_image, 0)
28
+ print(f'final shape: {input_image.shape}')
29
 
30
  # Get inference result
31
  result_infer = compiled_model([input_image])[output_layer]