Spaces:

oscared
/

mariposas_demo

Runtime error

oscared commited on Apr 19, 2023

Commit

787ecd5

1 Parent(s): 39cc8de

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -10,6 +10,6 @@ def carga_modelo(model_name='ceyda/butterfly_croppe_uniq1K_512',model_version=No
 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).deatch().cpu().numpy().asttype(np.uint8)
         return ims

 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 ims