Spaces:

Chancee12
/

Satellite-Segmentation-AI

Build error

Chancee12 commited on Apr 14, 2023

Commit

964f4ce

1 Parent(s): bc5c075

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -54,7 +54,7 @@ dice_loss = sm.losses.DiceLoss(class_weights=weights)
 focal_loss = sm.losses.CategoricalFocalLoss()
 total_loss = dice_loss + (1 * focal_loss)
-satellite_model = load_model('model/satellite_segmentation_full.h5',
                              custom_objects=({'dice_loss_plus_1focal_loss': total_loss, 'jaccard_coef': jaccard_coef}))

 focal_loss = sm.losses.CategoricalFocalLoss()
 total_loss = dice_loss + (1 * focal_loss)
+satellite_model = load_model('satellite_segmentation_full_v2.h5',
                              custom_objects=({'dice_loss_plus_1focal_loss': total_loss, 'jaccard_coef': jaccard_coef}))