Update app.py

app.py

@@ -54,9 +54,7 @@ model_pt = "mit5k_uegan_psnr_25.59.pth"
 # parse config file
 config = OmegaConf.load(CONFIG)
 
-config = dict2namespace(config)
-
-device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 config['train']['cuda_visible_device'] = device
 model = NamedCurves(config.model).to(device)