Spaces:

arnabdas8901
/

Magnifying_Lens

Running

Arnab Das commited on Nov 15, 2024

Commit

ffe0082

1 Parent(s): 736efbe

bug fix

Files changed (1) hide show

manipulate_model/utils.py CHANGED Viewed

@@ -19,7 +19,7 @@ def get_config_and_model(model_root="manipulate_model/demo-model/audio"):
         config.model.decoder = OmegaConf.load(config.model.decoder)
     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"])

         config.model.decoder = OmegaConf.load(config.model.decoder)
     model = Model(config=config)
+    model_file = hf_hub_download("arnabdas8901/manipulation_detection_transformer", filename= "weights.pt")
     print(model_file)
     #weights = torch.load(os.path.join(model_root, "weights.pt"))
     #model.load_state_dict(weights["model_state_dict"])