yashbyname commited on
Commit
fd7fd22
·
verified ·
1 Parent(s): b76fcf3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -74,7 +74,7 @@ def preprocess_image(image):
74
  #logger.info(f"Input image shape: {image.}, dtype: {image.dtype}")
75
 
76
  image = image.convert('RGB')
77
- image = image.resize((256, 256, 3))
78
  image = np.array(image)
79
 
80
  # Normalize pixel values
 
74
  #logger.info(f"Input image shape: {image.}, dtype: {image.dtype}")
75
 
76
  image = image.convert('RGB')
77
+ image = image.resize((256, 256))
78
  image = np.array(image)
79
 
80
  # Normalize pixel values