update

app.py

@@ -67,6 +67,11 @@ _answer_a = audio_vocabsize + 3
 _split = audio_vocabsize + 4
 
 
+def download_model(ckpt_dir):
+    repo_id = "gpt-omni/mini-omni"
+    snapshot_download(repo_id, local_dir=ckpt_dir, revision="main")
+
+
 if not os.path.exists(ckpt_dir):
     print(f"checkpoint directory {ckpt_dir} not found, downloading from huggingface")
     download_model(ckpt_dir)
@@ -88,11 +93,6 @@ model = model.to(device)
 model.eval()
 
 
-def download_model(ckpt_dir):
-    repo_id = "gpt-omni/mini-omni"
-    snapshot_download(repo_id, local_dir=ckpt_dir, revision="main")
-
-
 def get_input_ids_whisper_ATBatch(mel, leng, whispermodel, device):
     with torch.no_grad():
         mel = mel.unsqueeze(0).to(device)