oscared commited on
Commit
74074ff
1 Parent(s): 39cc8de

mariposa -> mariposas

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. utils.py +1 -0
app.py CHANGED
@@ -41,5 +41,5 @@ corre_boton = st.button(
41
  if ims is not None:
42
  cols = st.columns(n_mariposas)
43
  for j, im in enumerate(ims):
44
- i = j % n_mariposa
45
  cols[i].image(im, use_column_width=True)
 
41
  if ims is not None:
42
  cols = st.columns(n_mariposas)
43
  for j, im in enumerate(ims):
44
+ i = j % n_mariposas
45
  cols[i].image(im, use_column_width=True)
utils.py CHANGED
@@ -2,6 +2,7 @@ import numpy as np
2
  import torch
3
  from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
4
 
 
5
  def carga_modelo(model_name='ceyda/butterfly_croppe_uniq1K_512',model_version=None):
6
  gan = LightweightGAN.from_pretrained(model_name, vesion=model_version)
7
  gan.eval()
 
2
  import torch
3
  from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
4
 
5
+
6
  def carga_modelo(model_name='ceyda/butterfly_croppe_uniq1K_512',model_version=None):
7
  gan = LightweightGAN.from_pretrained(model_name, vesion=model_version)
8
  gan.eval()