diff --git a/deepface/models/facial_recognition/Buffalo_L.py b/deepface/models/facial_recognition/Buffalo_L.py index c86f315..efe1524 100644 --- a/deepface/models/facial_recognition/Buffalo_L.py +++ b/deepface/models/facial_recognition/Buffalo_L.py @@ -25,7 +25,8 @@ class Buffalo_L(FacialRecognition): raise ModuleNotFoundError( "InsightFace and its dependencies are optional for the Buffalo_L model. " "Please install them with: " - "pip install insightface>=0.7.3 onnxruntime>=1.9.0 typing-extensions pydantic albumentations" + "pip install insightface>=0.7.3 onnxruntime>=1.9.0 typing-extensions pydantic" + "albumentations" ) from err # Define the model filename and subdirectory @@ -86,7 +87,8 @@ class Buffalo_L(FacialRecognition): """ Extract face embedding from a pre-cropped face image. Args: - img: Preprocessed face image with shape (1, 112, 112, 3) or batch (batch_size, 112, 112, 3) + img: Preprocessed face image with shape (1, 112, 112, 3) + or batch (batch_size, 112, 112, 3) Returns: Face embedding as a list of floats or list of lists of floats """ @@ -98,4 +100,5 @@ class Buffalo_L(FacialRecognition): embedding = np.array(embedding).flatten() else: raise ValueError(f"Unexpected embedding type: {type(embedding)}") - return embedding.tolist() \ No newline at end of file + return embedding.tolist() + \ No newline at end of file