fix logic error

wenet-e2e · Nov 5, 2023 · 8ceb1d1 · 8ceb1d1
1 parent f00f358
commit 8ceb1d1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/wespeaker/cli/speaker.py b/wespeaker/cli/speaker.py
@@ -66,7 +66,10 @@ def extract_embedding(self, audio_path: str, apply_vad: bool = False):
     def compute_similarity(self, audio_path1: str, audio_path2) -> float:
         e1 = self.extract_embedding(audio_path1, True)
         e2 = self.extract_embedding(audio_path2, True)
-        return self.cosine_distance(e1, e2)
+        if e1 is None or e2 is None:
+            return 0.0
+        else:
+            return self.cosine_distance(e1, e2)
 
     def cosine_distance(self, e1, e2):
         return np.dot(e1, e2) / (norm(e1) * norm(e2))