breadlicker45 commited on
Commit
985349b
·
1 Parent(s): 327680d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ else:
37
  def generate(num_images, interpolate):
38
  if interpolate:
39
  z1 = torch.randn([1, G.z_dim])# latent codes
40
- z2 = torch.randn([1, G.z_dim])# latent codes
41
  zs = torch.cat([z1 + (z2 - z1) * i / (num_images-1) for i in range(num_images)], 0)
42
  else:
43
  zs = torch.randn([num_images, G.z_dim])# latent codes
 
37
  def generate(num_images, interpolate):
38
  if interpolate:
39
  z1 = torch.randn([1, G.z_dim])# latent codes
40
+ z2 = torch.randn([0, G.z_dim])# latent codes
41
  zs = torch.cat([z1 + (z2 - z1) * i / (num_images-1) for i in range(num_images)], 0)
42
  else:
43
  zs = torch.randn([num_images, G.z_dim])# latent codes