diff --git a/deepface/models/demography/Race.py b/deepface/models/demography/Race.py index 4537bed..cec6aaa 100644 --- a/deepface/models/demography/Race.py +++ b/deepface/models/demography/Race.py @@ -97,7 +97,7 @@ def load_model( # -------------------------- - race_model = Model(inputs=model.input, outputs=base_model_output) + race_model = Model(inputs=model.inputs, outputs=base_model_output) # --------------------------