Hugging Face's logo

sadhvikbathini
/
Densenet121-OSCC

Model card Files
xet
Community
sadhvikbathini commited on
Commit
eeb8bde
·
verified ·
1 Parent(s): 0164e2e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -1,8 +1,8 @@
1
  ---
2
  license: gpl-3.0
3
  ---
4
-
5
- model = torchvision.models.densenet121(pretrained=False)
6
  ---
 
 
7
  model.load_state_dict(torch.load('best_model.pth'))
8
- ---
 
1
  ---
2
  license: gpl-3.0
3
  ---
4
+ Usage:
 
5
  ---
6
+ model = torchvision.models.densenet121(pretrained=False)
7
+
8
  model.load_state_dict(torch.load('best_model.pth'))