Commit
·
5b40ecd
1
Parent(s):
e00fd58
Update app.py
Browse files
app.py
CHANGED
|
@@ -47,7 +47,7 @@ def dice_coef(y_true, y_pred, smooth=1):
|
|
| 47 |
return K.mean((2 * intersection + smooth) / (union + smooth), axis=0)
|
| 48 |
|
| 49 |
# Load the model
|
| 50 |
-
seg_model = tf.keras.models.load_model('seg_unet_model.h5', custom_objects={'Combo_loss': Combo_loss, 'dice_coef': dice_coef}
|
| 51 |
|
| 52 |
|
| 53 |
|
|
|
|
| 47 |
return K.mean((2 * intersection + smooth) / (union + smooth), axis=0)
|
| 48 |
|
| 49 |
# Load the model
|
| 50 |
+
seg_model = tf.keras.models.load_model('seg_unet_model.h5', custom_objects={'Combo_loss': Combo_loss, 'dice_coef': dice_coef})
|
| 51 |
|
| 52 |
|
| 53 |
|