Spaces:

AH-MD
/

Satellite-Imagery-Segmentation

Runtime error

Ahm-ed commited on Apr 5, 2023

Commit

1dc5526

1 Parent(s): a590d2b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,4 @@
 import os
-import cv2
 import random
 import numpy as np
 from PIL import Image
@@ -25,7 +24,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_weights/satellite_segmentation_full.h5', custom_objects=({'dice_loss_plus_1focal_loss': total_loss, 'jaccard_coef': jaccard_coef}))
 def process_input_image(image_source):

 import os
 import random
 import numpy as np
 from PIL import Image
 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}))
 def process_input_image(image_source):