Update app.py

app.py

@@ -125,7 +125,8 @@ def weighted_categorical_crossentropy(weights):
     return wcce
 
 # Load the model
-model = tf.keras.models.load_model("model_2.h5", custom_objects={"jacard":jacard, "wcce":weighted_categorical_crossentropy})
+#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})
 
 # Create a user interface for the model
 my_app = gr.Blocks()