Spaces:

ombui
/

Image_Prediction

Runtime error

ombui commited on May 28, 2023

Commit

0f0322a

1 Parent(s): fdd86b2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,8 +24,8 @@ dice_loss = sm.losses.DiceLoss(class_weights = weights)
 focal_loss = sm.losses.CategoricalFocalLoss()
 total_loss = dice_loss + (1 * focal_loss)
-satellite_model = load_model('C:/Users/sa/Desktop/Model_Training/satellite_segmentation_full.h5',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 = load_model('C:/Users/sa/Desktop/Model_Training/satellite_segmentation_full.h5', custom_objects={'dice_loss_plus_1focal_loss': total_loss})
 def process_input_image(image_source):
   image = np.expand_dims(image_source, 0)