Spaces:

rodragon737
/

mariposas-st

Runtime error

rodragon737 commited on Aug 5, 2023

Commit

11f5d1d

•

1 Parent(s): 6c12f8a

Update utils.py

Files changed (1) hide show

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)
     gan.eval()
     return gan