Hugging Face's logo

Spaces:
Kiwinicki
/
sat2map
Running

App Files Community
1
Kiwinicki commited on
Commit
9041310
·
verified ·
1 Parent(s): e49c48c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ with open(config_path, "r") as f:
23
  cfg = OmegaConf.create(config_dict)
24
 
25
  generator = Generator(cfg)
26
- generator.load_state_dict(torch.load(generator_path))
27
  generator.eval()
28
 
29
  from PIL import Image
 
23
  cfg = OmegaConf.create(config_dict)
24
 
25
  generator = Generator(cfg)
26
+ generator.load_state_dict(torch.load(generator_path, map_location=torch.device('cpu')))
27
  generator.eval()
28
 
29
  from PIL import Image