diff --git a/deepface/models/demography/Age.py b/deepface/models/demography/Age.py index f3a21a1..d9d08fe 100644 --- a/deepface/models/demography/Age.py +++ b/deepface/models/demography/Age.py @@ -89,7 +89,7 @@ def load_model( # -------------------------- - age_model = Model(inputs=model.input, outputs=base_model_output) + age_model = Model(inputs=model.inputs, outputs=base_model_output) # -------------------------- diff --git a/deepface/models/demography/Gender.py b/deepface/models/demography/Gender.py index 77f0ec1..f55c571 100644 --- a/deepface/models/demography/Gender.py +++ b/deepface/models/demography/Gender.py @@ -85,7 +85,7 @@ def load_model( # -------------------------- - gender_model = Model(inputs=model.input, outputs=base_model_output) + gender_model = Model(inputs=model.inputs, outputs=base_model_output) # --------------------------