Update app.py

app.py

@@ -58,7 +58,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 
 config['train']['cuda_visible_device'] = device
 model = NamedCurves(config.model, device=device).to(device)
-model.load_state_dict(torch.load(model_pt)["model_state_dict"])
+model.load_state_dict(torch.load(model_pt, map_location="cpu")["model_state_dict"])
 
 def load_img(filename, norm=True,):
     img = np.array(Image.open(filename).convert("RGB"))