Hugging Face's logo

Spaces:
huggan
/
huggingnft
Running

App Files Community
AlekseyKorshuk commited on
Commit
bba50e1
·
1 Parent(s): 730f0eb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ class Generator(nn.Module):
34
 
35
 
36
  model = Generator()
37
- weights_path = hf_hub_download('AlekseyKorshuk/dooggies', 'pytorch_model.bin')
38
  model.load_state_dict(torch.load(weights_path, map_location=torch.device('cpu')))
39
 
40
 
 
34
 
35
 
36
  model = Generator()
37
+ weights_path = hf_hub_download('huggingnft/dooggies', 'pytorch_model.bin')
38
  model.load_state_dict(torch.load(weights_path, map_location=torch.device('cpu')))
39
 
40