fradinho commited on
Commit
af0ffe6
·
1 Parent(s): 454e2ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -225,7 +225,7 @@ 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")
229
  #model.compile(optimizer='adam',
230
  # loss=[bce_dice, focal_iou],
231
  # metrics=metrics)
 
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", custom_objects={"jacard":jacard, "bce_dice":bce_dice, "focal_iou":focal_iou})
229
  #model.compile(optimizer='adam',
230
  # loss=[bce_dice, focal_iou],
231
  # metrics=metrics)