Agregar ims en return

utils.py

@@ -15,4 +15,4 @@ def genera(gan, batch_size=1):
     with torch.no_grad():
         ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp_(0.0, 1.0) * 255
         ims = ims.permute(0,2,3,1).detach().cpu().numpy().astype(np.uint8)
-    return
+    return ims