AfshinMA commited on
Commit
538fffb
·
verified ·
1 Parent(s): 9392d24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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()