Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,7 @@ app = FastAPI()
|
|
11 |
|
12 |
# Funci贸n para construir el modelo manualmente
|
13 |
def build_model():
|
|
|
14 |
model = tf.keras.Sequential(
|
15 |
[
|
16 |
tf.keras.layers.Input(
|
@@ -20,6 +21,11 @@ def build_model():
|
|
20 |
tf.keras.layers.Dense(1, activation="sigmoid", name="dense_3"),
|
21 |
]
|
22 |
)
|
|
|
|
|
|
|
|
|
|
|
23 |
model.load_weights(
|
24 |
"mimodelo.weights.h5"
|
25 |
) # Aseg煤rate de que los nombres de las capas coincidan para que los pesos se carguen correctamente
|
|
|
11 |
|
12 |
# Funci贸n para construir el modelo manualmente
|
13 |
def build_model():
|
14 |
+
"""
|
15 |
model = tf.keras.Sequential(
|
16 |
[
|
17 |
tf.keras.layers.Input(
|
|
|
21 |
tf.keras.layers.Dense(1, activation="sigmoid", name="dense_3"),
|
22 |
]
|
23 |
)
|
24 |
+
"""
|
25 |
+
model = tf.keras.Sequential([
|
26 |
+
tf.keras.layers.Dense(units=4, activation='sigmoid', input_shape=(2,)),
|
27 |
+
tf.keras.layers.Dense(units=1, activation='sigmoid')
|
28 |
+
])
|
29 |
model.load_weights(
|
30 |
"mimodelo.weights.h5"
|
31 |
) # Aseg煤rate de que los nombres de las capas coincidan para que los pesos se carguen correctamente
|