gabo152210 commited on
Commit
f9cfad3
·
verified ·
1 Parent(s): 7cc4f20

Upload 4 files

Browse files
Files changed (5) hide show
  1. .gitattributes +1 -0
  2. app.py +34 -0
  3. generator_full.keras +3 -0
  4. requirements.txt +4 -0
  5. runtime.txt +1 -0
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ generator_full.keras filter=lfs diff=lfs merge=lfs -text
app.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ Streamlit app – generates 1-10 MNIST-style digits using your trained cGAN
3
+ Run: streamlit run app.py
4
+ """
5
+
6
+ import streamlit as st
7
+ import tensorflow as tf
8
+ import numpy as np
9
+ from PIL import Image
10
+
11
+ LATENT_DIM = 100
12
+ NUM_CLASSES = 10
13
+ MODEL_FILE = "generator_full.keras" # <— same name you downloaded
14
+
15
+ # ---------- 1. Load generator only once per worker ----------
16
+ @st.cache_resource(show_spinner="Cargando modelo…")
17
+ def load_generator(model_path=MODEL_FILE):
18
+ # load_model includes architecture, so no need to rebuild by hand
19
+ return tf.keras.models.load_model(model_path, compile=False)
20
+
21
+ gen = load_generator()
22
+
23
+ # ---------- 2. Streamlit UI ----------
24
+ st.title("✍️ Generador de dígitos manuscritos (cGAN, 20 epochs)")
25
+ digit = st.number_input("Dígito (0-9)", min_value=0, max_value=9, value=4, step=1)
26
+ num = 5
27
+
28
+ if st.button("Generar"):
29
+ z = tf.random.normal([num, LATENT_DIM])
30
+ lbl = tf.constant([[digit]] * num)
31
+ imgs = (gen([z, lbl], training=False) + 1) / 2 # scale [-1,1] → [0,1]
32
+ cols = st.columns(num)
33
+ for c, img in zip(cols, imgs.numpy().squeeze()):
34
+ c.image(Image.fromarray((img * 255).astype("uint8"), "L"), use_column_width=True)
generator_full.keras ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4fd39f72321f992b73084050c57ea025760d4dfa0a3460d85fbd74db3bab1033
3
+ size 9386112
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ streamlit>=1.34.0
2
+ tensorflow-cpu>=2.17.0
3
+ numpy
4
+ Pillow
runtime.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ python-3.10.12