Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|