espejelomar commited on
Commit
c417ada
1 Parent(s): dda50ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -18
app.py CHANGED
@@ -3,26 +3,15 @@ import streamlit as st
3
  from utils import carga_modelo, genera
4
 
5
  ## Página principal
6
- st.title("Butterfly GAN (GAN de mariposas)")
7
- st.write(
8
- "Modelo Light-GAN entrenado con 1000 imágenes de mariposas tomadas de la colección del Museo Smithsonian."
9
- )
10
 
11
  ## Barra lateral
12
- st.sidebar.subheader("¡Esta mariposa no existe! Ni en América Latina 🤯.")
13
  st.sidebar.image("assets/logo.png", width=200)
14
- st.sidebar.caption(
15
- f"[Modelo](https://huggingface.co/ceyda/butterfly_cropped_uniq1K_512) y [Dataset](https://huggingface.co/datasets/huggan/smithsonian_butterflies_subset) usados."
16
- )
17
- st.sidebar.caption(f"*Disclaimers:*")
18
- st.sidebar.caption(
19
- "* Este demo es una versión simplificada del creado por [Ceyda Cinarel](https://github.com/cceyda) y [Jonathan Whitaker](https://datasciencecastnet.home.blog/) ([link](https://huggingface.co/spaces/huggan/butterfly-gan)) durante el hackathon [HugGan](https://github.com/huggingface/community-events). Cualquier error se atribuye a [Omar Espejel](https://twitter.com/espejelomar)."
20
- )
21
- st.sidebar.caption(
22
- "* Modelo basado en el [paper](https://openreview.net/forum?id=1Fqg133qRaI) *Towards Faster and Stabilized GAN Training for High-fidelity Few-shot Image Synthesis*."
23
- )
24
 
25
- ## Cargamos modelo
26
  repo_id = "ceyda/butterfly_cropped_uniq1K_512"
27
  version_modelo = "57d36a15546909557d9f967f47713236c8288838"
28
  modelo_gan = carga_modelo(repo_id, version_modelo)
@@ -30,8 +19,6 @@ modelo_gan = carga_modelo(repo_id, version_modelo)
30
  ## Generamos 4 mariposas
31
  n_mariposas = 4
32
 
33
-
34
-
35
  def corre():
36
  with st.spinner("Generando, espero un poco..."):
37
  ims = genera(modelo_gan, n_mariposas)
3
  from utils import carga_modelo, genera
4
 
5
  ## Página principal
6
+ st.title("Generador de mariposas")
7
+ st.write("Este es un modelo Light GAN entrenado y utilizado con Platzi")
 
 
8
 
9
  ## Barra lateral
10
+ st.sidebar.subheader("!Esta mariposa no existe, ¿puedes creerlo?")
11
  st.sidebar.image("assets/logo.png", width=200)
12
+ st.sidebar.caption("Demo creado en vivo.")
 
 
 
 
 
 
 
 
 
13
 
14
+ ## Cargamos el modelo
15
  repo_id = "ceyda/butterfly_cropped_uniq1K_512"
16
  version_modelo = "57d36a15546909557d9f967f47713236c8288838"
17
  modelo_gan = carga_modelo(repo_id, version_modelo)
19
  ## Generamos 4 mariposas
20
  n_mariposas = 4
21
 
 
 
22
  def corre():
23
  with st.spinner("Generando, espero un poco..."):
24
  ims = genera(modelo_gan, n_mariposas)