Jason Adrian commited on
Commit
e8376fa
1 Parent(s): d14fd28

changes to app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from resnet18 import ResNet18
7
 
8
  model = ResNet18(1, 5)
9
 
10
- checkpoint = torch.load('C:\jason\semester 8\Magang\Hugging-face-bodypartxr\bodypartxr\acc=0.94.ckpt')
11
 
12
  # The state dict will contains net.layer_name
13
  # Our model doesn't contains `net.` so we have to rename it
 
7
 
8
  model = ResNet18(1, 5)
9
 
10
+ checkpoint = torch.load('acc=0.94.ckpt')
11
 
12
  # The state dict will contains net.layer_name
13
  # Our model doesn't contains `net.` so we have to rename it