miittnnss commited on
Commit
be12e7f
1 Parent(s): c24888c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ netG.eval()
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()
 
39
 
40
  def generate_image():
41
  with torch.no_grad():
42
+ noise = torch.randn(1, 100, 1, 1)
43
  fake_image = netG(noise)
44
 
45
  generated_image = fake_image.squeeze().cpu().numpy()