app.py

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)