Abdu07
/

multitask-model

Image Classification

Model card Files Files and versions Community

Abdu07 commited on Mar 27

Commit

9f54355

·

verified ·

1 Parent(s): 9bce084

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -45,7 +45,7 @@ model = MultiTaskModel(...)
 # Load the saved state dictionary (trained weights)
-model.load_state_dict(torch.load("Yolloplusclassproject_weights.pth", map_location="cpu"))
 model.eval()
 ```
 Alternatively, you can test the model directly via our interactive demo:

 # Load the saved state dictionary (trained weights)
+model.load_state_dict(torch.load("DualSight.pth", map_location="cpu"))
 model.eval()
 ```
 Alternatively, you can test the model directly via our interactive demo: