Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ def classify_image(input_image):
|
|
| 24 |
|
| 25 |
# Resize the input image to the expected size
|
| 26 |
input_image = input_image.reshape((1, 224, 224, 3)) # Reshape for a single prediction
|
| 27 |
-
input_image = keras.applications.mobilenet_v2.preprocess_input(input_image)
|
| 28 |
|
| 29 |
# Perform prediction
|
| 30 |
prediction = inception_net.predict(input_image).flatten()
|
|
|
|
| 24 |
|
| 25 |
# Resize the input image to the expected size
|
| 26 |
input_image = input_image.reshape((1, 224, 224, 3)) # Reshape for a single prediction
|
| 27 |
+
input_image = tf.keras.applications.mobilenet_v2.preprocess_input(input_image)
|
| 28 |
|
| 29 |
# Perform prediction
|
| 30 |
prediction = inception_net.predict(input_image).flatten()
|