Spaces:
Paused
Paused
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
|