bug fix

manipulate_model/utils.py

@@ -20,10 +20,9 @@ def get_config_and_model(model_root="manipulate_model/demo-model/audio"):
 
     model = Model(config=config)
     model_file = hf_hub_download("arnabdas8901/manipulation_detection_transformer", filename= "weights.pt")
-    model = model._load_as_pickle(model, model_file, map_location="cpu", strict=True)
+    weights = torch.load(model_file,  map_location=torch.device("cpu"))
+    model.load_state_dict(weights["model_state_dict"])
     print("### Model loaded from :", model_file)
-    #weights = torch.load(os.path.join(model_root, "weights.pt"))
-    #model.load_state_dict(weights["model_state_dict"])
 
     return config, model