windy2612 commited on
Commit
7a785fd
·
verified ·
1 Parent(s): b39f433

Update app.py

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