Spaces:

srjosueaaron
/

streamlit

Sleeping

Josue Aaron Soriano Rivero commited on Aug 23

Commit

ab5c1d1

•

1 Parent(s): 2114012

detach

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -48,5 +48,5 @@ def carga_modelo(nombre_modelo="ceyda/butterfly_cropped_uniq1K_512", model_versi
 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 ims