fradinho commited on
Commit
9a8a333
·
1 Parent(s): 27a73a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -135,7 +135,8 @@ from tensorflow.python.keras.utils import generic_utils
135
  # Load the model
136
  #model = tf.keras.models.load_model("model.h5", custom_objects={"jacard":jacard, "wcce":weighted_categorical_crossentropy})
137
  #model = tf.keras.models.load_model("model_2.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
138
- model = tf.keras.models.load_model("model_2_A.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
 
139
 
140
 
141
  # Create a user interface for the model
 
135
  # Load the model
136
  #model = tf.keras.models.load_model("model.h5", custom_objects={"jacard":jacard, "wcce":weighted_categorical_crossentropy})
137
  #model = tf.keras.models.load_model("model_2.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
138
+ ###model = tf.keras.models.load_model("model_2_A.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
139
+ model = tf.keras.models.load_model("model_2_A_(1).h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
140
 
141
 
142
  # Create a user interface for the model