Hugging Face's logo

Spaces:
swapniel99
/
cifar10
Sleeping

App Files Community
swapniel99 commited on
Commit
8973e5a
·
1 Parent(s): d1fdeca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ missed_df['predicted_vals'] = missed_df['predicted_vals'].map(cifar10.classes)
22
  missed_df = missed_df.sample(frac=1)
23
 
24
  model = Model(cifar10)
25
- model.load_state_dict(torch.load('S12_model.pth', map_location='mps'))
26
  model.eval()
27
 
28
  transform = transforms.Compose([
 
22
  missed_df = missed_df.sample(frac=1)
23
 
24
  model = Model(cifar10)
25
+ model.load_state_dict(torch.load('S12_model.pth', map_location='cpu'))
26
  model.eval()
27
 
28
  transform = transforms.Compose([