update

app_locally.py

@@ -6,8 +6,8 @@ import gradio as gr
 from melo.api import TTS
 
 # Init EN/ZH baseTTS and ToneConvertor
-from openvoice import se_extractor
-from openvoice.api import ToneColorConverter
+from OpenVoice import se_extractor
+from OpenVoice.api import ToneColorConverter
 import devicetorch
 
 print(f"openvoice = {dir(openvoice)}")
@@ -23,7 +23,7 @@ def predict(prompt, style, audio_file_pth, mic_file_path, use_mic, language):
     text_hint = ''
 
     tts_model = TTS(language=language, device=device)
-    speaker_id = models[language].hps.data.spk2id
+    speaker_id = tts_model.hps.data.spk2id
     speaker_key = speaker_key.lower().replace('_', '-')
     source_se = torch.load(f'checkpoints/base_speakers/ses/{speaker_key}.pth', map_location=device)