Spaces:
Sleeping
Sleeping
KeniBrandonGM
commited on
Commit
•
855aaea
1
Parent(s):
0f17eae
corrigendo error en función genera
Browse files
utils.py
CHANGED
@@ -10,5 +10,5 @@ def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=
|
|
10 |
def genera(gan, batch_size=1):
|
11 |
with torch.no_grad():
|
12 |
ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp_(0.0,1.0)*255
|
13 |
-
ims = ims.permute(0,2,3,1).
|
14 |
return ims
|
|
|
10 |
def genera(gan, batch_size=1):
|
11 |
with torch.no_grad():
|
12 |
ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp_(0.0,1.0)*255
|
13 |
+
ims = ims.permute(0,2,3,1).detach().cpu().numpy().astype(np.uint8)
|
14 |
return ims
|