Spaces:

Davies
/

Streamlit_Demo

Runtime error

Davies commited on Mar 22, 2023

Commit

53bc0ec

1 Parent(s): be13489

actualizacion de model a 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(modelo_name ="ceyda/butterfly_cropped_uniq1K_512", model_version =None):
-    gan = LightweightGAN.from_pretrained(model_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_name, version =model_version)
     gan.eval()
     return gan