Spaces:

Chancee12
/

Satellite-Segmentation-AI

Build error

Chancee12 commited on Apr 14, 2023

Commit

f22bd91

1 Parent(s): 9272eaa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -77,7 +77,7 @@ dice_loss = sm.losses.DiceLoss(class_weights = weights)
 focal_loss = sm.losses.CategoricalFocalLoss()
 total_loss = dice_loss + (1 * focal_loss)
-satellite_model = tf.keras.models.load_model("satellite_segmentation_v3_full/saved_model.pb", custom_objects=({'dice_loss_plus_1focal_loss': total_loss, 'jaccard_coef': jaccard_coef}))
 def process_input_image(image_source):
     image = np.expand_dims(image_source, 0)

 focal_loss = sm.losses.CategoricalFocalLoss()
 total_loss = dice_loss + (1 * focal_loss)
+satellite_model = tf.keras.models.load_model("satellite_segmentation_v3_full", custom_objects=({'dice_loss_plus_1focal_loss': total_loss, 'jaccard_coef': jaccard_coef}))
 def process_input_image(image_source):
     image = np.expand_dims(image_source, 0)