Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = '
|
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 |
|