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