bonifaciohansel commited on
Commit
674a2d1
1 Parent(s): 9e66935

Se corrigen errores

Browse files
__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).deatch().cpu().numpy().astype(np.uint8)
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