refactor(prediction): remove bbox feature removal logic

refactor(vectorizer): filter out background polygons during vectorization

predictor/app.py

@@ -98,14 +98,6 @@ def predict(
     if remove_metadata:
         shutil.rmtree(base_path)
 
-    if (
-        "features" in prediction_geojson_data
-        and len(prediction_geojson_data["features"]) > 0
-    ):
-        prediction_geojson_data["features"].pop(
-            -1
-        )  # Remove the last feature explicitly for the removal of whole bbox
-
     for feature in prediction_geojson_data["features"]:
         feature["properties"]["building"] = "yes"
         feature["properties"]["source"] = "fAIr"

predictor/vectorizer.py

@@ -57,11 +57,16 @@ def vectorize(
         resampling=Resampling.nearest,
     )
 
-    # Close raster files after merging
     for raster in rasters:
         raster.close()
 
-    polygons = [shape(s) for s, _ in shapes(mosaic, transform=output)]
+    polygons = []
+
+    for s, value in shapes(mosaic, transform=output):
+
+        if value != 0:  # value 0 is background
+            polygons.append(shape(s))
+
     gs = gpd.GeoSeries(polygons, crs=kwargs["crs"])
 
     # Explode MultiPolygons