Spaces:

espejelomar
/

mariposas

Runtime error

espejelomar commited on Jul 26, 2022

Commit

dc809e2

•

1 Parent(s): 9a080f3

arreglar error

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -10,5 +10,5 @@ def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=
 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().astype(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