sharktide commited on
Commit
3d2a6b5
·
verified ·
1 Parent(s): ea16666

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -58,7 +58,7 @@ def preprocess_image(image_file, model):
58
  # Convert image to numpy array
59
  image = np.array(image)
60
 
61
- if (model == "fruitbot0" or model == "fruitbot-expanded):
62
  image = cv2.resize(image, (240, 240))
63
  image = image.reshape(-1, 240, 240, 3)
64
  elif model == "fruitbot1":
 
58
  # Convert image to numpy array
59
  image = np.array(image)
60
 
61
+ if (model == "fruitbot0" or model == "fruitbot-expanded"):
62
  image = cv2.resize(image, (240, 240))
63
  image = image.reshape(-1, 240, 240, 3)
64
  elif model == "fruitbot1":