Update app.py

app.py

@@ -413,7 +413,7 @@ def main():
     device_id = opt.gpu_id
     device = torch.device('cuda:%d' % device_id if torch.cuda.is_available() else 'cpu')
     opt.device = device
-
+    print(device)
 
     # load model
     model = build_models(opt, edit_config=edit_config)