From e4a43fd86bbb764f45d0ae28ccd2ae17bddd0fab Mon Sep 17 00:00:00 2001 From: Sefik Ilkin Serengil Date: Mon, 18 Dec 2023 09:31:54 +0000 Subject: [PATCH] Update DeepFace.py functional may come from keras or tf.keras. hard to manage in typing. used any for build_model. --- deepface/DeepFace.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py index 5989a2e..40eb664 100644 --- a/deepface/DeepFace.py +++ b/deepface/DeepFace.py @@ -7,7 +7,6 @@ import pickle import logging from typing import Any, Dict, List, Tuple, Union from keras.engine.functional import Functional -from deepface.basemodels import DlibResNet, SFace # 3rd party dependencies import numpy as np @@ -48,7 +47,7 @@ if tf_version == 2: # ----------------------------------- -def build_model(model_name: str) -> Union[Functional, DlibResNet.DlibResNet, SFace.SFaceModel]: +def build_model(model_name: str) -> Any: """ This function builds a deepface model Parameters: