Rafiahmed Patel commited on
Commit
b4c00af
·
1 Parent(s): 2eb6fda

new changes

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ from faster_whisper import WhisperModel
7
  import torch
8
  import torchaudio as ta
9
  import torchaudio.transforms as transforms
10
- from chatterbox import ChatterboxMultilingualTTS
11
  import logging
12
  from typing import List, Dict
13
  from deep_translator import GoogleTranslator
@@ -96,7 +96,7 @@ def generate_translated_audio(
96
  # Generate audio for this segment
97
  wav = tts_model.generate(
98
  segment['translated_text'],
99
- target_language,
100
  audio_prompt_path=reference_audio_path,
101
  exaggeration=0.2,
102
  cfg_weight=0.8,
 
7
  import torch
8
  import torchaudio as ta
9
  import torchaudio.transforms as transforms
10
+ from chatterbox.mtl_tts import ChatterboxMultilingualTTS, SUPPORTED_LANGUAGES
11
  import logging
12
  from typing import List, Dict
13
  from deep_translator import GoogleTranslator
 
96
  # Generate audio for this segment
97
  wav = tts_model.generate(
98
  segment['translated_text'],
99
+ language_id = target_language,
100
  audio_prompt_path=reference_audio_path,
101
  exaggeration=0.2,
102
  cfg_weight=0.8,