fradinho commited on
Commit
0377b31
·
1 Parent(s): c36a57b

Update app.py

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