bug fix

manipulate_model/utils.py

@@ -3,6 +3,7 @@ import torch
 import torchaudio
 import numpy as np
 from omegaconf import OmegaConf
+from huggingface_hub import hf_hub_download
 from torch.nn.functional import pad, normalize, softmax
 
 from manipulate_model.model import Model
@@ -17,7 +18,9 @@ def get_config_and_model(model_root="manipulate_model/demo-model/audio"):
     if isinstance(config.model.decoder, str):
         config.model.decoder = OmegaConf.load(config.model.decoder)
 
-    model = Model._from_pretrained("arnabdas8901/manipulation_detection_transformer", config=config)
+    model = Model(config=config)
+    model_file = hf_hub_download("arnabdas8901/manipulation_detection_transformer")
+    print(model_file)
     #weights = torch.load(os.path.join(model_root, "weights.pt"))
     #model.load_state_dict(weights["model_state_dict"])