Spaces:

fradinho
/

drone_footage_segmentation

Runtime error

fradinho commited on Oct 27, 2023

Commit

a0a1c9d

1 Parent(s): 26893ff

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -204,9 +204,9 @@ 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})
 # Create a user interface for the model

 # 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})
 # Create a user interface for the model