windy2612 commited on
Commit
1041d2e
·
verified ·
1 Parent(s): c80b9ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ with open(config_path, 'r') as data:
26
  config = yaml.safe_load(data)
27
  system = System(config)
28
  checkpoint_path = 'weights/parseq.ckpt'
29
- checkpoint = torch.load(checkpoint_path, map_location = 'cuda')
30
  system.load_state_dict(checkpoint['state_dict'])
31
  system.to(device)
32
 
 
26
  config = yaml.safe_load(data)
27
  system = System(config)
28
  checkpoint_path = 'weights/parseq.ckpt'
29
+ checkpoint = torch.load(checkpoint_path, map_location = 'cpu')
30
  system.load_state_dict(checkpoint['state_dict'])
31
  system.to(device)
32