Hugging Face's logo

Spaces:
Sijuade
/
S12
Sleeping

App Files Community
Sijuade commited on
Commit
dd4195b
•
1 Parent(s): 00e8269

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ from dataset.dataset import *
19
 
20
 
21
  model = ResNet18(20, None)
22
- model = model.load_from_checkpoint("/content/S12/logs/lightning_logs/version_1/checkpoints/epoch=119-step=23520.ckpt", map_location=torch.device("cpu"))
23
  model.eval()
24
 
25
  dataloader_args = dict(shuffle=True, batch_size=64)
 
19
 
20
 
21
  model = ResNet18(20, None)
22
+ model = model.load_from_checkpoint("resnet18.ckpt", map_location=torch.device("cpu"))
23
  model.eval()
24
 
25
  dataloader_args = dict(shuffle=True, batch_size=64)