Spaces:

Sijuade
/

S12

Runtime error

Sijuade commited on Aug 4, 2023

Commit

64b6db1

1 Parent(s): 0c7ae04

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ from dataset.dataset import *
 model = ResNet18(20, None)
-model = model.load_from_checkpoint("/content/drive/MyDrive/ERAV1/S12/resnet18.ckpt", map_location=torch.device("cpu"))
 dataloader_args = dict(shuffle=True, batch_size=64)
 _, test_transforms = get_transforms(mu, std)

 model = ResNet18(20, None)
+model = model.load_from_checkpoint("resnet18.ckpt", map_location=torch.device("cpu"))
 dataloader_args = dict(shuffle=True, batch_size=64)
 _, test_transforms = get_transforms(mu, std)