gabo152210 commited on
Commit
42f2be9
·
verified ·
1 Parent(s): 4ecdd41

Upload 3 files

Browse files
Files changed (4) hide show
  1. .gitattributes +1 -0
  2. app.py +35 -0
  3. generator_full.keras +3 -0
  4. requirements.txt +4 -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,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import tensorflow as tf
3
+ import numpy as np
4
+ from pathlib import Path
5
+ from PIL import Image
6
+
7
+ LATENT_DIM = 100
8
+ MODEL_FILE = Path(__file__).with_name("generator_full.keras")
9
+
10
+ _gen = None # cargamos “lazy” para arrancar rápido
11
+ def get_generator():
12
+ global _gen
13
+ if _gen is None:
14
+ _gen = tf.keras.models.load_model(MODEL_FILE, compile=False)
15
+ return _gen
16
+
17
+ def generate(digit: int):
18
+ z = tf.random.normal([5, LATENT_DIM])
19
+ lbl = tf.constant([[digit]] * 5)
20
+ imgs = (get_generator()([z, lbl], training=False) + 1) / 2 # [0,1]
21
+ return [
22
+ Image.fromarray((img.numpy() * 255).astype("uint8").squeeze(), mode="L")
23
+ for img in imgs
24
+ ]
25
+
26
+ demo = gr.Interface(
27
+ fn=generate,
28
+ inputs=gr.Number(label="Digit 0-9", precision=0, value=4),
29
+ outputs=gr.Gallery(label="Five samples", columns=5, rows=1),
30
+ title="Hand-written Digit Generator (cGAN · 20 epochs)",
31
+ description="Pick a digit and get five MNIST-style images."
32
+ )
33
+
34
+ if __name__ == "__main__":
35
+ demo.launch()
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
+ gradio==3.40.0 # UI framework
2
+ tensorflow-cpu==2.16.1 # cp312 manylinux_2_34 → compatible con Spaces
3
+ numpy
4
+ Pillow