szili2011 commited on
Commit
f46e6ef
1 Parent(s): 7ccec0f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from keras.models import load_model
4
  from keras.preprocessing.image import load_img, img_to_array
5
 
6
  # Load the pre-trained model
7
- model = load_model('your_model_path.h5')
8
 
9
  # Define class names for Boy or Girl classification
10
  class_names = ['Boy', 'Girl']
 
4
  from keras.preprocessing.image import load_img, img_to_array
5
 
6
  # Load the pre-trained model
7
+ model = load_model('best_model.h5')
8
 
9
  # Define class names for Boy or Girl classification
10
  class_names = ['Boy', 'Girl']