Spaces:
Sleeping
Sleeping
File size: 1,106 Bytes
6e7834e 3a2c0fa 6e7834e 26b4a28 6e7834e 26b4a28 6e7834e 26b4a28 6e7834e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
import streamlit as st
from utils import carga_modelo, genera
##pagina principal
st.title("Generador de mariposas")
st.write("Esto es un modelo Light GAN entrenado y utilizado con Platzi")
## barra lateral
st.sidebar.subheader("esta mariposa no existe! Puedes creerlo??")
st.sidebar.image("asset/logo.png", width=200)
st.sidebar.caption("Demo creado en un curso de Platzi")
## Cargamos el modelo
repo_id = "https://huggingface.co/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(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_buton = st.button(
"Genera mariposas porfa",
on_click= corre,
help="Estamos en vuelo, abrocha el cinturón."
)
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_sidth=True) |