Update app.py

app.py

@@ -32,7 +32,7 @@ def get_model(config):
 
 def load(model, file, device=None, strict=True):
     if device is None: device = 'cpu'
-    elif isinstance(device, int): device = torch.device('cuda', device)
+    elif isinstance(device, int): device = torch.device('cuda', 0)
     assert os.path.isfile(file)
     state = torch.load(file, map_location=device)
     if set(state.keys()) == {'model', 'opt'}: