Spaces:
Sleeping
Sleeping
ronalleiva
commited on
Commit
•
1251f27
1
Parent(s):
0282bab
Update utils.py
Browse filesCambios como esta documentado el uso de ceyda/butterfly
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"
|
8 |
-
gan = LightweightGAN.from_pretrained(model_name
|
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 |
|