Spaces:

joacorf33
/

butterflies

Runtime error

Joaquin Romero Flores commited on Mar 20, 2023

Commit

a098cb6

1 Parent(s): c243d27

fixing spelling bug

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -23,5 +23,5 @@ def generate(gan, batch_size=1):
         # Cleaning process to properly fit it to the model
         ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp_(0.0, 1.0) * 255
         #
-        ims = ims.permute(0, 2, 3, 1).deatch().cpu().numpy().astype(np.uint8)
     return ims

         # Cleaning process to properly fit it to the model
         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 ims