Hugging Face's logo

Spaces:
miittnnss
/
dcgan-image-generator
Runtime error

App Files Community
3
miittnnss commited on
Commit
e3004b1
•
1 Parent(s): bfd4e48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ class Generator(nn.Module):
33
  netG = Generator()
34
 
35
  device = "cuda" if torch.cuda.is_available() else "cpu"
36
- model_file = "generator.pth"
37
  netG.load_state_dict(torch.load(model_file, map_location=device))
38
  netG.eval()
39
 
 
33
  netG = Generator()
34
 
35
  device = "cuda" if torch.cuda.is_available() else "cpu"
36
+ model_file = "model.pth"
37
  netG.load_state_dict(torch.load(model_file, map_location=device))
38
  netG.eval()
39