Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ def load_model(selected_model):
61
  model = Model(inputs=base_model.input, outputs=x)
62
 
63
  # Compile model (ensure it matches training settings)
64
- model.compile(optimizer=tf.keras.optimizers.RMSprop(lr=1e-4),
65
  loss='binary_crossentropy',
66
  metrics=['accuracy'])
67
 
 
61
  model = Model(inputs=base_model.input, outputs=x)
62
 
63
  # Compile model (ensure it matches training settings)
64
+ model.compile(optimizer=tf.keras.optimizers.Adam(lr=1e-4),
65
  loss='binary_crossentropy',
66
  metrics=['accuracy'])
67