fradinho commited on
Commit
42e343f
·
1 Parent(s): 582d497

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -225,10 +225,10 @@ from tensorflow.python.keras.utils import generic_utils
225
  #model = tf.keras.models.load_model("model_2.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
226
  ###model = tf.keras.models.load_model("model_2_A.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
227
  #model = tf.keras.models.load_model("model_2_A_0.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
228
- model = tf.keras.models.load_model("model_3_A.h5", compile ="False")
229
- model.compile(optimizer='adam',
230
- loss=[bce_dice, focal_iou],
231
- metrics=metrics)
232
 
233
  # Create a user interface for the model
234
  my_app = gr.Blocks()
 
225
  #model = tf.keras.models.load_model("model_2.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
226
  ###model = tf.keras.models.load_model("model_2_A.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
227
  #model = tf.keras.models.load_model("model_2_A_0.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
228
+ model = tf.keras.models.load_model("model_3_A.h5")
229
+ #model.compile(optimizer='adam',
230
+ # loss=[bce_dice, focal_iou],
231
+ # metrics=metrics)
232
 
233
  # Create a user interface for the model
234
  my_app = gr.Blocks()