mjms commited on
Commit
c5edc1a
1 Parent(s): 7561fe3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ dice_loss = sm.losses.DiceLoss(class_weights=weights)
29
  focal_loss = sm.losses.CategoricalFocalLoss()
30
  total_loss = dice_loss + (1 * focal_loss)
31
 
32
- model_path = 'models/satellite_segmentation_100-epochs.h5'
33
  saved_model = load_model(model_path,
34
  custom_objects=({'dice_loss_plus_1focal_loss': total_loss,
35
  'jaccard_coef': jaccard_coef}))
 
29
  focal_loss = sm.losses.CategoricalFocalLoss()
30
  total_loss = dice_loss + (1 * focal_loss)
31
 
32
+ model_path = 'model/satellite_segmentation_100-epochs.h5'
33
  saved_model = load_model(model_path,
34
  custom_objects=({'dice_loss_plus_1focal_loss': total_loss,
35
  'jaccard_coef': jaccard_coef}))