Update app.py

app.py

@@ -16,11 +16,18 @@ from keras import backend as K
 from keras.models import load_model
 
 # Replace 'your_username' and 'your_model_name' with the appropriate values
+import zipfile
+
 model_identifier = "Chancee12/satellite_segmentation_v3_full"
-model_directory = hf_hub_download(repo_id=model_identifier, repo_type="model")
-satellite_model = tf.keras.models.load_model(model_directory, custom_objects=({'dice_loss_plus_1focal_loss' : total_loss, 'jaccard_coef': jaccard_coef}))
+filename = "satellite_segmentation_v3_full.zip"
+model_directory = hf_hub_download(repo_id=model_identifier, filename=filename, repo_type="model")
+
+# Extract the zip file
+with zipfile.ZipFile(model_directory, 'r') as zip_ref:
+    zip_ref.extractall("satellite_segmentation_v3_full")
+
 
-# The rest of your code remains the same
+# The rest of code remains the same
 
 class_building = '#2A2A2A'
 class_building = class_building.lstrip('#')