Spaces:
Sleeping
Sleeping
Upload 3 files
Browse files- .gitattributes +1 -0
- app.py +35 -0
- generator_full.keras +3 -0
- 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
|