update

app_locally.py

@@ -6,12 +6,16 @@ import gradio as gr
 from melo.api import TTS
 
 # Init EN/ZH baseTTS and ToneConvertor
-from OpenVoice import se_extractor
-from OpenVoice.api import BaseSpeakerTTS, ToneColorConverter
+from openVoice import se_extractor
+from openVoice.api import ToneColorConverter
 import devicetorch
 
 device = devicetorch.get(torch)
 
+ckpt_converter = 'checkpoints/converter'
+tone_color_converter = ToneColorConverter(f'{ckpt_converter}/config.json', device=device)
+tone_color_converter.load_ckpt(f'{ckpt_converter}/checkpoint.pth')
+
 def predict(prompt, style, audio_file_pth, mic_file_path, use_mic, language):
     # initialize a empty info
     text_hint = ''