Spaces:
Running
Running
Update app.py
Browse files
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('
|
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']
|