Merge branch 'main' of https://huggingface.co/spaces/oscared/mariposas_demo

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 
-from utils import carga_model, genera
+from utils import carga_modelo, genera
 
 ##pagina principla
 st.title('Generador de mariposas')
@@ -14,7 +14,7 @@ st.sidebar.caption('Demo creado en vivo.')
 
 ## cargamos el model
 repo_id = 'ceyda/butterfly_cropped_uniq1K_512'
-modelo_gan = carga_model(repo_id)
+modelo_gan = carga_modelo(repo_id)
 
 
 ## genera 4 mariposas

utils.py

@@ -11,6 +11,6 @@ def carga_modelo(model_name='ceyda/butterfly_croppe_uniq1K_512',model_version=No
 def genera(gan, batch_size=1):
     with torch.no_grad():
         ims = gan.G(torch.randn(batch_size, gan.latent_dim).clamp_(0.0,1.0)*255)
-        ims = ims.permute(0,2,3,1).deatch().cpu().numpy().asttype(np.uint8)
+        ims = ims.permute(0,2,3,1).detach().cpu().numpy().astype(np.uint8)
         return ims