Spaces:
Build error
Build error
File size: 1,055 Bytes
c700b9c |
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 43 44 |
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)
|