Spaces:

Davies
/

Streamlit_Demo

Runtime error

Davies commited on Mar 22, 2023

Commit

9068b7f

1 Parent(s): 53bc0ec

Update utils.py

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(modelo_name ="ceyda/butterfly_cropped_uniq1K_512", model_version =None):
-    gan = LightweightGAN.from_pretrained(modelo_name_name, version =model_version)
     gan.eval()
     return gan

 from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
 def carga_modelo(modelo_name ="ceyda/butterfly_cropped_uniq1K_512", model_version =None):
+    gan = LightweightGAN.from_pretrained(modelo_name, version =model_version)
     gan.eval()
     return gan