RandomCatLover commited on
Commit
bd4d9f6
·
1 Parent(s): 8c62505
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ with open(f'{model_folder}/labels.txt', 'r') as f:
25
  # model = tf.saved_model.load(f'{model_folder}/last_layer.hdf5')
26
  model = tf.keras.models.load_model(f'{model_folder}/last_layer.hdf5')
27
  # %%
28
- def classify_image(model, inp):
29
  inp = cv2.resize(inp, (224,224,))
30
  inp = inp.reshape((-1, 224, 224, 3))
31
  inp = tf.keras.applications.mobilenet_v2.preprocess_input(inp)
 
25
  # model = tf.saved_model.load(f'{model_folder}/last_layer.hdf5')
26
  model = tf.keras.models.load_model(f'{model_folder}/last_layer.hdf5')
27
  # %%
28
+ def classify_image(inp):
29
  inp = cv2.resize(inp, (224,224,))
30
  inp = inp.reshape((-1, 224, 224, 3))
31
  inp = tf.keras.applications.mobilenet_v2.preprocess_input(inp)