Update app.py

app.py

@@ -36,8 +36,8 @@ else:
 
 def generate(num_images, interpolate):
     if interpolate:
-        z1 = torch.randn([0, G.z_dim])# latent codes
-        z2 = torch.randn([0, G.z_dim])# latent codes
+        z1 = torch.randn([1, G.z_dim])# latent codes
+        z2 = torch.randn([1, G.z_dim])# latent codes
         zs = torch.cat([z1 + (z2 - z1) * i / (num_images-1) for i in range(num_images)], 0)
     else:
         zs = torch.randn([num_images, G.z_dim])# latent codes