RaulHuarote commited on
Commit
0b7cee6
verified
1 Parent(s): 88d69a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
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