Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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",
|
|
|
|
|
|
|
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()
|