import streamlit as st from utils import carga_modelo, genera st.title("Generador de mariposas") st.write("Este es un modelo Light GAN entrenado y utilizado con Platzi") st.sidebar.subheader("Esta mariposa es creada artificialmente, no es real") st.sidebar.image("assets/logo.png", width=200) st.sidebar.caption("Demo creado en vivo.") repo_id = "ceyda/butterfly_cropped_uniq1K_512" modelo_gan = carga_modelo(repo_id) n_mariposas = 4 def corre(): with st.spinner("Generando, Espera"): ims = genera(modelo_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", on_click=corre, help="Estamos en vuelo, mira atentamente" ) 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)