diff --git a/deepface/modules/modeling.py b/deepface/modules/modeling.py index f9c6464..271fc25 100644 --- a/deepface/modules/modeling.py +++ b/deepface/modules/modeling.py @@ -40,7 +40,7 @@ def build_model(model_name: str) -> Any: if not "model_obj" in globals(): model_obj = {} - if not model_name in model_obj: + if not model_name in model_obj.keys(): model = models.get(model_name) if model: model = model()