Spaces:
Running
Running
Update voice.py
Browse files
voice.py
CHANGED
@@ -255,8 +255,17 @@ class vits:
|
|
255 |
return audio
|
256 |
|
257 |
|
|
|
|
|
258 |
class TTS:
|
259 |
def __init__(self, voice_obj, voice_speakers):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
260 |
self._voice_obj = voice_obj
|
261 |
self._voice_speakers = voice_speakers
|
262 |
self._strength_dict = {"x-weak": 0.25, "weak": 0.5, "Medium": 0.75, "Strong": 1, "x-strong": 1.25}
|
@@ -274,17 +283,6 @@ class TTS:
|
|
274 |
except Exception as e:
|
275 |
self.logger.warning(f"Load DIMENSIONAL_EMOTION_MODEL failed {e}")
|
276 |
|
277 |
-
# Initialization information
|
278 |
-
self.logger = logging.getLogger("vits-simple-api")
|
279 |
-
self.logger.info(f"torch:{torch.__version__} cuda_available:{torch.cuda.is_available()}")
|
280 |
-
self.logger.info(f'device:{device} device.type:{device.type}')
|
281 |
-
if self._vits_speakers_count != 0: self.logger.info(f"[VITS] {self._vits_speakers_count} speakers")
|
282 |
-
if self._hubert_speakers_count != 0: self.logger.info(f"[hubert] {self._hubert_speakers_count} speakers")
|
283 |
-
if self._w2v2_speakers_count != 0: self.logger.info(f"[w2v2] {self._w2v2_speakers_count} speakers")
|
284 |
-
self.logger.info(f"{self._speakers_count} speakers in total")
|
285 |
-
if self._speakers_count == 0:
|
286 |
-
self.logger.warning(f"No model was loaded")
|
287 |
-
|
288 |
@property
|
289 |
def voice_speakers(self):
|
290 |
return self._voice_speakers
|
|
|
255 |
return audio
|
256 |
|
257 |
|
258 |
+
import logging
|
259 |
+
|
260 |
class TTS:
|
261 |
def __init__(self, voice_obj, voice_speakers):
|
262 |
+
# ロガーを初期化
|
263 |
+
self.logger = logging.getLogger(__name__)
|
264 |
+
self.logger.setLevel(logging.WARNING)
|
265 |
+
handler = logging.StreamHandler()
|
266 |
+
handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
|
267 |
+
self.logger.addHandler(handler)
|
268 |
+
|
269 |
self._voice_obj = voice_obj
|
270 |
self._voice_speakers = voice_speakers
|
271 |
self._strength_dict = {"x-weak": 0.25, "weak": 0.5, "Medium": 0.75, "Strong": 1, "x-strong": 1.25}
|
|
|
283 |
except Exception as e:
|
284 |
self.logger.warning(f"Load DIMENSIONAL_EMOTION_MODEL failed {e}")
|
285 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
286 |
@property
|
287 |
def voice_speakers(self):
|
288 |
return self._voice_speakers
|