Update app.py

app.py

@@ -9,7 +9,7 @@ import numpy as np
 
 import ChatTTS
 
-import OpenVoice.se_extractor as se_extractor
+from OpenVoice import se_extractor
 from OpenVoice.api import ToneColorConverter
 import soundfile
 
@@ -68,14 +68,11 @@ def chat_tts(text, temperature, top_P, top_K, audio_seed_input, text_seed_input,
         return text_data
 
 # OpenVoice Clone
-ckpt_converter_en = 'checkpoints/converter'
+ckpt_converter = 'checkpoints/converter'
 device = 'cuda:0'
 
-#device = "cpu"
-
-tone_color_converter = ToneColorConverter(f'{ckpt_converter_en}/config.json', device=device)
-tone_color_converter.load_ckpt(f'{ckpt_converter_en}/checkpoint.pth')
-
+tone_color_converter = ToneColorConverter(f'{ckpt_converter}/config.json', device=device)
+tone_color_converter.load_ckpt(f'{ckpt_converter}/checkpoint.pth')
 
 def generate_audio(text, audio_ref, temperature, top_P, top_K, audio_seed_input, text_seed_input, refine_text_flag, refine_text_input):
     save_path = "output.wav"