Spaces:

fradinho
/

drone_footage_segmentation

Runtime error

fradinho commited on Oct 27, 2023

Commit

5349b13

1 Parent(s): a0a1c9d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -203,8 +203,8 @@ def weighted_categorical_crossentropy(weights):
 from tensorflow.python.keras.utils import generic_utils
 # Load the model
-#model = tf.keras.models.load_model("model.h5", custom_objects={"jacard":jacard, "wcce":weighted_categorical_crossentropy})
-model = tf.keras.models.load_model("model_2.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
 ###model = tf.keras.models.load_model("model_2_A.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
 #model = tf.keras.models.load_model("model_2_A_0.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})

 from tensorflow.python.keras.utils import generic_utils
 # Load the model
+model = tf.keras.models.load_model("model.h5", custom_objects={"jacard":jacard, "wcce":weighted_categorical_crossentropy})
+#model = tf.keras.models.load_model("model_2.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
 ###model = tf.keras.models.load_model("model_2_A.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
 #model = tf.keras.models.load_model("model_2_A_0.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})