Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -39,7 +39,7 @@ netG.eval()
|
|
39 |
|
40 |
def generate_image():
|
41 |
with torch.no_grad():
|
42 |
-
noise = torch.randn(1,
|
43 |
fake_image = netG(noise)
|
44 |
|
45 |
generated_image = fake_image.squeeze().cpu().numpy()
|
|
|
39 |
|
40 |
def generate_image():
|
41 |
with torch.no_grad():
|
42 |
+
noise = torch.randn(1, 128, 1, 1)
|
43 |
fake_image = netG(noise)
|
44 |
|
45 |
generated_image = fake_image.squeeze().cpu().numpy()
|