demfier commited on
Commit
3588e50
·
1 Parent(s): 31ee107

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ model = tf.keras.layers.Dropout(rate=0.1)(model)
12
  model = tf.keras.layers.Dense(2, activation='sigmoid')(model)
13
  model = tf.keras.models.Model(inputs=efficient_net.input, outputs=model)
14
 
15
- model.compile(loss='binary_crossentropy',optimizer = 'Adam', metrics= ['accuracy'])
16
 
17
  model.load_weights('./checkpoint')
18
 
 
12
  model = tf.keras.layers.Dense(2, activation='sigmoid')(model)
13
  model = tf.keras.models.Model(inputs=efficient_net.input, outputs=model)
14
 
15
+ model.compile(loss='binary_crossentropy',optimizer =tf.keras.optimizers.legacy.Adam(), metrics= ['accuracy'])
16
 
17
  model.load_weights('./checkpoint')
18