fradinho commited on
Commit
582d497
·
1 Parent(s): 5a73aef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -225,7 +225,10 @@ from tensorflow.python.keras.utils import generic_utils
225
  #model = tf.keras.models.load_model("model_2.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
226
  ###model = tf.keras.models.load_model("model_2_A.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
227
  #model = tf.keras.models.load_model("model_2_A_0.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
228
- model = tf.keras.models.load_model("model_3_A.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice, "focal_iou":focal_iou})
 
 
 
229
 
230
  # Create a user interface for the model
231
  my_app = gr.Blocks()
 
225
  #model = tf.keras.models.load_model("model_2.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
226
  ###model = tf.keras.models.load_model("model_2_A.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
227
  #model = tf.keras.models.load_model("model_2_A_0.h5", custom_objects={"jacard":jacard, "bce_dice":bce_dice})
228
+ model = tf.keras.models.load_model("model_3_A.h5", compile ="False")
229
+ model.compile(optimizer='adam',
230
+ loss=[bce_dice, focal_iou],
231
+ metrics=metrics)
232
 
233
  # Create a user interface for the model
234
  my_app = gr.Blocks()