Pijush2023 commited on
Commit
53bd549
·
verified ·
1 Parent(s): b814546

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -568,9 +568,10 @@ def generate_audio_mars5(text):
568
 
569
  for chunk in chunks:
570
  wav = torch.zeros(1, mars5.sr) # Use a placeholder silent audio for the reference
571
- cfg = config_class(**kwargs_dict)
572
  ar_codes, wav_out = mars5.tts(chunk, wav, "", cfg=cfg)
573
 
 
574
  temp_audio_path = os.path.join(tempfile.gettempdir(), f"mars5_audio_{len(audio_segments)}.wav")
575
  torchaudio.save(temp_audio_path, wav_out.unsqueeze(0), mars5.sr)
576
  audio_segments.append(AudioSegment.from_wav(temp_audio_path))
 
568
 
569
  for chunk in chunks:
570
  wav = torch.zeros(1, mars5.sr) # Use a placeholder silent audio for the reference
571
+ cfg = config_class(**{k: kwargs_dict[k] for k in kwargs_dict if k in config_class.__dataclass_fields__})
572
  ar_codes, wav_out = mars5.tts(chunk, wav, "", cfg=cfg)
573
 
574
+
575
  temp_audio_path = os.path.join(tempfile.gettempdir(), f"mars5_audio_{len(audio_segments)}.wav")
576
  torchaudio.save(temp_audio_path, wav_out.unsqueeze(0), mars5.sr)
577
  audio_segments.append(AudioSegment.from_wav(temp_audio_path))