Spaces:

miittnnss
/

dcgan-image-generator

Runtime error

miittnnss commited on Sep 26, 2023

Commit

a266672

•

1 Parent(s): ce5a8ff

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,8 @@ class Generator(nn.Module):
 netG = Generator()
 device = "cuda" if torch.cuda.is_available() else "cpu"
-netG.load_state_dict(torch.load("dcgan.pth", map_location=device))
 netG.eval()
 def generate_image():

 netG = Generator()
 device = "cuda" if torch.cuda.is_available() else "cpu"
+model_file = "generator.pth"
+netG.load_state_dict(torch.load(model_file, map_location=device))
 netG.eval()
 def generate_image():