Hugging Face's logo

Abdu07
/
multitask-model

Image Classification
Model card Files
xet
Community
Abdu07 commited on
Commit
9f54355
·
verified ·
1 Parent(s): 9bce084

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -45,7 +45,7 @@ model = MultiTaskModel(...)
45
 
46
 
47
  # Load the saved state dictionary (trained weights)
48
- model.load_state_dict(torch.load("Yolloplusclassproject_weights.pth", map_location="cpu"))
49
  model.eval()
50
  ```
51
  Alternatively, you can test the model directly via our interactive demo:
 
45
 
46
 
47
  # Load the saved state dictionary (trained weights)
48
+ model.load_state_dict(torch.load("DualSight.pth", map_location="cpu"))
49
  model.eval()
50
  ```
51
  Alternatively, you can test the model directly via our interactive demo: