yuragoithf commited on
Commit
f504885
·
1 Parent(s): 3b78676

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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}, '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")
 
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")