Spaces:
Runtime error
Runtime error
Commit
·
90fa634
1
Parent(s):
59d9031
Made Changes
Browse files
app.py
CHANGED
|
@@ -25,10 +25,10 @@ args.cx = '772f0ccc480364116'
|
|
| 25 |
|
| 26 |
|
| 27 |
# model
|
| 28 |
-
device = '
|
| 29 |
model = get_model(args)
|
| 30 |
model.to(device)
|
| 31 |
-
checkpoint = torch.hub.load_state_dict_from_url(args.resume, map_location='
|
| 32 |
model.load_state_dict(checkpoint['model'], strict=True)
|
| 33 |
|
| 34 |
|
|
|
|
| 25 |
|
| 26 |
|
| 27 |
# model
|
| 28 |
+
device = 'cpu' #torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 29 |
model = get_model(args)
|
| 30 |
model.to(device)
|
| 31 |
+
checkpoint = torch.hub.load_state_dict_from_url(args.resume, map_location='cpu')
|
| 32 |
model.load_state_dict(checkpoint['model'], strict=True)
|
| 33 |
|
| 34 |
|