Spaces:

KeniBrandonGM
/

MariposasStr2

Sleeping

KeniBrandonGM commited on Aug 20, 2023

Commit

0f17eae

1 Parent(s): dbf2a60

corriengo error en funcion carga modelo

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
 def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=None):
-    gan = LightweightGAN.from_pretrained(model_name, vesion=model_version)
     gan.eval()
     return gan

 from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
 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