Hugging Face's logo

Spaces:
brxerq
/
Bird_Classification
Sleeping

App Files Community
brxerq commited on
Commit
7875725
·
verified ·
1 Parent(s): fdd5879

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from tensorflow.keras.models import load_model
7
  import cv2
8
 
9
  # Load the pre-trained model
10
- model = load_model('bird_model.h5')
11
 
12
  # Load class labels from your text file
13
  train_info = []
 
7
  import cv2
8
 
9
  # Load the pre-trained model
10
+ model = tf.keras.models.load_model('bird_model4.h5', custom_objects={'KerasLayer': hub.KerasLayer})
11
 
12
  # Load class labels from your text file
13
  train_info = []