Faysal4200 commited on
Commit
a2bbb5a
·
verified ·
1 Parent(s): 87999d7

Added resize to main app

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -151,6 +151,7 @@ full_names = {
151
  }
152
 
153
  def preprocess_image(image):
 
154
  if image.dtype != np.uint8:
155
  image = cv2.normalize(image, None, 0, 255, cv2.NORM_MINMAX).astype(np.uint8)
156
  lab = cv2.cvtColor(image, cv2.COLOR_RGB2LAB)
 
151
  }
152
 
153
  def preprocess_image(image):
154
+ image = cv2.resize(image, (224, 224))
155
  if image.dtype != np.uint8:
156
  image = cv2.normalize(image, None, 0, 255, cv2.NORM_MINMAX).astype(np.uint8)
157
  lab = cv2.cvtColor(image, cv2.COLOR_RGB2LAB)