ronalleiva commited on
Commit
1251f27
1 Parent(s): 0282bab

Update utils.py

Browse files

Cambios como esta documentado el uso de ceyda/butterfly

Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -4,8 +4,8 @@ from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
4
 
5
 
6
  ## Cargamos el modelo desde el Hub de Hugging Face
7
- def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=None):
8
- gan = LightweightGAN.from_pretrained(model_name, version=model_version)
9
  gan.eval()
10
  return gan
11
 
 
4
 
5
 
6
  ## Cargamos el modelo desde el Hub de Hugging Face
7
+ def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512"):
8
+ gan = LightweightGAN.from_pretrained(model_name)
9
  gan.eval()
10
  return gan
11