Spaces:

venkat-natchi
/

resnet18_cifar10

Sleeping

venkyyuvy commited on Aug 6, 2023

Commit

3cb2ba0

1 Parent(s): 7e9b1a3

map_location to cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ from matplotlib import pyplot as plt
 resnet_18_model = ResNet18()
-resnet_18_model.load_state_dict(torch.load('resnet18.pth'))
 resnet_18_model.eval()
 classes = ('plane', 'car', 'bird', 'cat', 'deer',

 resnet_18_model = ResNet18()
+resnet_18_model.load_state_dict(torch.load('resnet18.pth', map_location='cpu'))
 resnet_18_model.eval()
 classes = ('plane', 'car', 'bird', 'cat', 'deer',