update space

app.py

@@ -54,14 +54,14 @@ from ola.utils import disable_torch_init
 from ola.datasets.preprocess import tokenizer_image_token, tokenizer_speech_image_token, tokenizer_speech_question_image_token
 from ola.mm_utils import get_model_name_from_path, KeywordsStoppingCriteria, process_anyres_video, process_anyres_highres_image_genli
 from ola.constants import IGNORE_INDEX, DEFAULT_IMAGE_TOKEN, IMAGE_TOKEN_INDEX, DEFAULT_SPEECH_TOKEN
-from ola.CosyVoice.cosyvoice.cli.cosyvoice import CosyVoice
+from ola.CosyVoice-main.cosyvoice.cli.cosyvoice import CosyVoice
 
 model_path = "THUdyh/Ola-7b"
 tokenizer, model, image_processor, _ = load_pretrained_model(model_path, None)
 model = model.to('cuda').eval()
 model = model.bfloat16()
 
-# tts_model = CosyVoice('./ola/CosyVoice/pretrained_models/CosyVoice-300M-SFT', load_jit=True, load_onnx=False, fp16=True)
+tts_model = CosyVoice('FunAudioLLM/CosyVoice-300M-SFT', load_jit=True, load_onnx=False, fp16=True)
 OUTPUT_SPEECH = False
 
 USE_SPEECH=False