Spaces:
Build error
Build error
| import streamlit as st | |
| from utils import carga_modelo,genera | |
| ## Pagina principal | |
| st.title("Generador de mariposas") | |
| st.write("Este es un modelo Light GAN entrenando y utilizado con Platzi")( | |
| ## Barra lateral | |
| st.sidebar.subheader("Esta mariposa no existe, ¿Puedes creeerlo?") | |
| st.sidebar.image("assets/logo.png",width=200) | |
| st.sidebar.caption("Demo creado en vivo.") | |
| ## Cargamos el modelo | |
| repo_id="ceyda/butterfly_cropped_uniq1K_512" | |
| modelo_gan=carga_modelo(repo_id) | |
| ##Generamos 4 mariposas | |
| n_mariposas=4 | |
| def corre(): | |
| with st.spinner("Generando,espera un poco"): | |
| ims=genera(model_gan,n_mariposas) | |
| st.session_state("ims")=ims | |
| if "ims" not in st.session_state: | |
| st.session_state("ims")=None | |
| corre() | |
| ims=st.session_state("ims") | |
| corre_boton=st.button( | |
| "Genera mariposas porfa", | |
| on_click=corre | |
| help="Estamos en vuelo,abrocha tu cinturon" | |
| ) | |
| if ims is not None: | |
| cols=st.columns(n_mariposas) | |
| for j,im in enumerate(ims): | |
| i=j%n_mariposas | |
| cols[i].image(im,use_column_width=True) | |