update space

app.py

@@ -56,6 +56,20 @@ from ola.mm_utils import get_model_name_from_path, KeywordsStoppingCriteria, pro
 from ola.constants import IGNORE_INDEX, DEFAULT_IMAGE_TOKEN, IMAGE_TOKEN_INDEX, DEFAULT_SPEECH_TOKEN
 from ola.CosyVoice_main.cosyvoice.cli.cosyvoice import CosyVoice
 
+from huggingface_hub import hf_hub_download
+
+whisper_path = hf_hub_download(
+    repo_id="THUdyh/Ola-7b",
+    filename="large-v3.pt",
+    local_dir="./"
+)
+
+beats_path = hf_hub_download(
+    repo_id="THUdyh/Ola-7b",
+    filename="BEATs_iter3_plus_AS2M_finetuned_on_AS2M_cpt2.pt",
+    local_dir="./"
+)
+
 model_path = "THUdyh/Ola-7b"
 tokenizer, model, image_processor, _ = load_pretrained_model(model_path, None)
 model = model.to('cuda').eval()