Spaces:

Segizu
/

butterflies_GAN

Sleeping

Segizu commited on Feb 11

Commit

fb57428

1 Parent(s): 58b9b37

token: None

Files changed (2) hide show

__pycache__/utils.cpython-39.pyc ADDED Viewed

Binary file (927 Bytes). View file

utils.py CHANGED Viewed

@@ -4,7 +4,7 @@ from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
 ## Cargamos el modelo desde el Hub de Hugging Face
 def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=None):
-    gan = LightweightGAN.from_pretrained(model_name, version=model_version, use_auth_token=True
     gan.eval()
     return gan

 ## Cargamos el modelo desde el Hub de Hugging Face
 def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=None):
+    gan = LightweightGAN.from_pretrained(model_name, version=model_version, use_auth_token=None)
     gan.eval()
     return gan