Spaces:
Sleeping
Sleeping
Update app.py
#1
by
EdwinLH
- opened
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.
|
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 |
|