yuragoithf
commited on
Commit
·
f504885
1
Parent(s):
3b78676
Update app.py
Browse files
app.py
CHANGED
@@ -50,7 +50,7 @@ def focal_loss_fixed(y_true, y_pred, gamma=2.0, alpha=0.25):
|
|
50 |
return focal_loss_fixed
|
51 |
|
52 |
# Load the model
|
53 |
-
seg_model = keras.models.load_model('seg_unet_model.h5', custom_objects={'Combo_loss': Combo_loss, 'focal_loss_fixed': focal_loss_fixed
|
54 |
|
55 |
# inputs = gr.inputs.Image(type="pil", label="Upload an image")
|
56 |
# image_output = gr.outputs.Image(type="pil", label="Output Image")
|
|
|
50 |
return focal_loss_fixed
|
51 |
|
52 |
# Load the model
|
53 |
+
seg_model = keras.models.load_model('seg_unet_model.h5', custom_objects={'Combo_loss': Combo_loss, 'focal_loss_fixed': focal_loss_fixed, 'dice_coef': dice_coef})
|
54 |
|
55 |
# inputs = gr.inputs.Image(type="pil", label="Upload an image")
|
56 |
# image_output = gr.outputs.Image(type="pil", label="Output Image")
|