Spaces:
Sleeping
Sleeping
David Chavarria
commited on
Commit
•
6677ec8
1
Parent(s):
df1ec88
Arreglando todo
Browse files- packages.txt +1 -0
- utils.py +1 -1
packages.txt
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
libgl1
|
utils.py
CHANGED
@@ -15,4 +15,4 @@ def genera(gan, batch_size=1):
|
|
15 |
with torch.no_grad():
|
16 |
ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp_(0.0, 1.0) * 255
|
17 |
ims = ims.permute(0, 2, 3, 1).detach().cpu().numpy().astype(np.uint8)
|
18 |
-
return ims
|
|
|
15 |
with torch.no_grad():
|
16 |
ims = gan.G(torch.randn(batch_size, gan.latent_dim)).clamp_(0.0, 1.0) * 255
|
17 |
ims = ims.permute(0, 2, 3, 1).detach().cpu().numpy().astype(np.uint8)
|
18 |
+
return ims
|