fradinho commited on
Commit
0df7dd0
·
1 Parent(s): b977c05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -127,7 +127,7 @@ def weighted_categorical_crossentropy(weights):
127
  # Load the model
128
  #model = tf.keras.models.load_model("model.h5", custom_objects={"jacard":jacard, "wcce":weighted_categorical_crossentropy})
129
  #model = tf.keras.models.load_model("model_2.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
130
- model = tf.keras.models.load_model("model_2_A (2).h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
131
 
132
  # Create a user interface for the model
133
  my_app = gr.Blocks()
 
127
  # Load the model
128
  #model = tf.keras.models.load_model("model.h5", custom_objects={"jacard":jacard, "wcce":weighted_categorical_crossentropy})
129
  #model = tf.keras.models.load_model("model_2.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
130
+ model = tf.keras.models.load_model("model_2_A (1).h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
131
 
132
  # Create a user interface for the model
133
  my_app = gr.Blocks()