bug fix

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import models as MOD
 import process_data as PD
 from transformers import pipeline
+from huggingface_hub import hf_hub_download
 
 model_master = {
     "SSL-AASIST (Trained on ASV-Spoof5)": {"eer_threshold": 3.3330237865448,
@@ -18,7 +19,8 @@ model_master = {
 }
 
 model = MOD.Model(None, "cpu")
-model.load_state_dict(torch.load("ssl_aasist_epoch_7.pth", map_location="cpu"))
+base_model_file = hf_hub_download("arnabdas8901/aasist-trained-asvspoof2024", filename="ssl_aasist_epoch_7.pth")
+model.load_state_dict(torch.load(base_model_file, map_location="cpu"))
 model.eval()
 loaded_model = "SSL-AASIST (Trained on ASV-Spoof5)"
 
@@ -28,7 +30,8 @@ def process(file, type):
     inp = getattr(PD, model_master[type]["data_process_func"])(file)
     if not loaded_model == type:
         model = getattr(MOD, model_master[type]["model_class"])(None, "cpu")
-        model.load_state_dict(torch.load(model_master[type]["model_checkpoint"], map_location="cpu"))
+        model_file = hf_hub_download("arnabdas8901/aasist-trained-asvspoof2024", filename=model_master[type]["model_checkpoint"])
+        model.load_state_dict(torch.load(model_file, map_location="cpu"))
         model.eval()
         loaded_model = type
 

orig_aasist_epoch_1.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f26ad87bca3d47e97ecfeac6fee6fcae93f62673a484d447c081d45911e3a027
-size 1276136

ssl_aasist_epoch_7.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:07e888c07dce17c15c0a299aaacc81ac90f8ad43a9daf9396234baebc44b1944
-size 1271618222