Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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":
|