download models from a model repo

app.py

@@ -14,6 +14,13 @@
 
 import gradio as gr
 import spaces
+from huggingface_hub import snapshot_download
+
+# Download models
+snapshot_download(
+    repo_id = "fffiloni/SALMONN-7B-PACK",
+    local_dir = "./models"
+)
 
 import argparse
 from model import SALMONN
@@ -25,10 +32,10 @@ class ff:
 
 parser = argparse.ArgumentParser()
 parser.add_argument("--device", type=str, default="cuda:0")
-parser.add_argument("--ckpt_path", type=str, default="./salmonn_7b_v0.pth")
-parser.add_argument("--whisper_path", type=str, default="./whisper_large_v2")
-parser.add_argument("--beats_path", type=str, default="./beats/BEATs_iter3_plus_AS2M_finetuned_on_AS2M_cpt2.pt")
-parser.add_argument("--vicuna_path", type=str, default="./vicuna-7b-v1.5")
+parser.add_argument("--ckpt_path", type=str, default="./models/salmonn_7b_v0.pth")
+parser.add_argument("--whisper_path", type=str, default="./models/whisper_large_v2")
+parser.add_argument("--beats_path", type=str, default="./models/beats/BEATs_iter3_plus_AS2M_finetuned_on_AS2M_cpt2.pt")
+parser.add_argument("--vicuna_path", type=str, default="./models/vicuna-7b-v1.5")
 parser.add_argument("--low_resource", action='store_true', default=False)
 parser.add_argument("--port", default=9527)