Spaces:

Abdu07
/

DualSight-Demo

Running

Abdu07 commited on Mar 27

Commit

9d8932c

verified ·

1 Parent(s): 938bc8e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ model = MultiTaskModel(resnet, feature_dim, num_obj_classes)
 model.to(device)
 repo_id = "Abdu07/multitask-model"
-filename = "Yolloplusclassproject_weights.pth"
 weights_path = hf_hub_download(repo_id=repo_id, filename=filename)
 state_dict = torch.load(weights_path, map_location="cpu")
 model.load_state_dict(state_dict)

 model.to(device)
 repo_id = "Abdu07/multitask-model"
+filename = "DualSight.pth"
 weights_path = hf_hub_download(repo_id=repo_id, filename=filename)
 state_dict = torch.load(weights_path, map_location="cpu")
 model.load_state_dict(state_dict)