soiz commited on
Commit
c4e984f
·
verified ·
1 Parent(s): 426f4f5

Update voice.py

Browse files
Files changed (1) hide show
  1. voice.py +9 -11
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