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()
|