Update app.py

app.py

@@ -7,7 +7,7 @@ use_gpu = True if torch.cuda.is_available() else False
 model = torch.hub.load('facebookresearch/pytorch_GAN_zoo:hub', 'DCGAN', pretrained=True, useGPU=use_gpu)
 
 def dcgan(num_images):
-  noise, _ = model.buildNoiseData(num_images)
+  noise, _ = model.buildNoiseData(int(num_images))
   with torch.no_grad():
       generated_images = model.test(noise)
   plt.imshow(torchvision.utils.make_grid(generated_images).permute(1, 2, 0).cpu().numpy())