Hugging Face's logo

Spaces:
avanish07
/
Crowd_Count
Sleeping

App Files Community
avanish07 commited on
Commit
963d428
·
1 Parent(s): f326c04

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,7 +67,7 @@ def make_layers(cfg, in_channels = 3,batch_norm=False,dilation = False):
67
 
68
  # Load the CSRNet model
69
  csrmodel = CSRNet(load_weights=True).cpu()
70
- checkpoint = torch.load("model.pt")
71
  csrmodel.load_state_dict(checkpoint)
72
  csrmodel.eval()
73
 
 
67
 
68
  # Load the CSRNet model
69
  csrmodel = CSRNet(load_weights=True).cpu()
70
+ checkpoint = torch.load("model.pt", map_location=torch.device('cpu'))
71
  csrmodel.load_state_dict(checkpoint)
72
  csrmodel.eval()
73