From fb43a178a38b88ec6942ba97770eead3a8de6955 Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Mon, 8 Jan 2024 16:22:32 +0900
Subject: [PATCH] Fix bug: add missing get_style_vector function

---
 common/tts_model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/common/tts_model.py b/common/tts_model.py
index f83a806..307c6c3 100644
--- a/common/tts_model.py
+++ b/common/tts_model.py
@@ -61,9 +61,9 @@ class Model:
     def get_style_vector_from_audio(
         self, audio_path: str, weight: float = 1.0
     ) -> np.ndarray:
-        from style_gen import extract_style_vector
+        from style_gen import get_style_vector
 
-        xvec = extract_style_vector(audio_path)
+        xvec = get_style_vector(audio_path)
         mean = self.style_vectors[0]
         xvec = mean + (xvec - mean) * weight
         return xvec