Update app.py

app.py

@@ -19,7 +19,7 @@ from huggingface_hub import snapshot_download
 # Download models
 snapshot_download(
     repo_id = "fffiloni/SALMONN-7B-PACK",
-    local_dir = "./models"
+    local_dir = "./"
 )
 
 import argparse
@@ -32,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="./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("--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("--low_resource", action='store_true', default=False)
 parser.add_argument("--port", default=9527)