Update app.py

app.py

@@ -131,6 +131,8 @@ def weighted_categorical_crossentropy(weights):
         return bce_dice(y_true, y_pred) * K.sum(y_true * Kweights, axis=-1)
     return wcce
 
+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})