Update app.py

app.py

@@ -95,8 +95,8 @@ opt.seed = -1
 
 ### Global variables and data structures
 
-device = f'cuda:{opt.device}' if opt.device >= 0 else 'cpu'
-
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+print(device)
 
 if opt.model is None:
     model_dict = {