Update app.py

app.py

@@ -18,8 +18,8 @@ from dataset.dataset import *
 
 
 
-model = ResNet18(20, None)
-model = model.load_from_checkpoint("resnet18.ckpt", map_location=torch.device("cpu"))
+# model = ResNet18(20, None)
+model = ResNet18.load_from_checkpoint("resnet18.ckpt", map_location=torch.device("cpu"))
 
 dataloader_args = dict(shuffle=True, batch_size=64)
 _, test_transforms = get_transforms(mu, std)