Update utils.py

utils.py

@@ -1,11 +1,11 @@
 import numpy as np
 import torch
-##from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
-from lightweight_gan import LightweightGAN
+from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
+##from 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)
+    gan = LightweightGAN.from_pretrained(model_name, version=model_version, use_auth_token)
     gan.eval()
     return gan