Hugging Face's logo

Spaces:
RandomCatLover
/
plants_disease
Sleeping

App Files Community
RandomCatLover commited on
Commit
5ea968c
·
1 Parent(s): d55ab02

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -35,8 +35,8 @@ with open(f'{model_folder}/labels.txt', 'r') as f:
35
  labels = f.read().split('\n')
36
 
37
  # model = tf.saved_model.load(f'{model_folder}/last_layer.hdf5')
38
- # model = tf.keras.models.load_model(f'{model_folder}/last_layer.hdf5')
39
- model = tf.keras.models.load_model(f'{model_folder}/MobileNetV2_last_layer.hdf5')
40
  # %%
41
  def classify_image(inp):
42
  inp = cv2.resize(inp, (224,224,))
 
35
  labels = f.read().split('\n')
36
 
37
  # model = tf.saved_model.load(f'{model_folder}/last_layer.hdf5')
38
+ model = tf.keras.models.load_model(f'{model_folder}/last_layer.hdf5')
39
+ #model = tf.keras.models.load_model(f'{model_folder}/MobileNetV2_last_layer.hdf5')
40
  # %%
41
  def classify_image(inp):
42
  inp = cv2.resize(inp, (224,224,))