Spaces:
Runtime error
Runtime error
bonifaciohansel
commited on
Commit
•
674a2d1
1
Parent(s):
9e66935
Se corrigen errores
Browse files- __pycache__/utils.cpython-37.pyc +0 -0
- assets/logo.jfif +0 -0
- assets/logo.png +0 -0
- utils.py +2 -2
__pycache__/utils.cpython-37.pyc
ADDED
Binary file (911 Bytes). View file
|
|
assets/logo.jfif
DELETED
Binary file (7.62 kB)
|
|
assets/logo.png
ADDED
utils.py
CHANGED
@@ -9,6 +9,6 @@ def carga_modelo(model_name = "ceyda/butterfly_cropped_uniq1K_512",model_version
|
|
9 |
|
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
|
|
|
9 |
|
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
|