Spaces:
Running
Running
Update app.py
Browse files
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 = '
|
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}))
|