Spaces:

rriverar75
/

mariposas

Runtime error

rriverar75 commited on Aug 13, 2023

Commit

cdf9525

1 Parent(s): 9509acd

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -5,7 +5,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)
     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=True)
     gan.eval()
     return gan