Return type correction

app.py

@@ -109,22 +109,28 @@ def generate_geojson(response):
 # Generate static map image
 @spaces.GPU
 def generate_static_map(geojson_data):
+    # Create a static map object with specified dimensions
     m = StaticMap(500, 500)
+    
+    # Process each feature in the GeoJSON
     for feature in geojson_data["features"]:
         geom_type = feature["geometry"]["type"]
         coords = feature["geometry"]["coordinates"]
 
         if geom_type == "Point":
+            # Add a blue marker for Point geometries
             m.add_marker(CircleMarker((coords[0], coords[1]), 'blue', 10))
         elif geom_type in ["MultiPoint", "LineString"]:
+            # Add a red marker for each point in MultiPoint or LineString geometries
             for coord in coords:
                 m.add_marker(CircleMarker((coord[0], coord[1]), 'red', 10))
         elif geom_type in ["Polygon", "MultiPolygon"]:
+            # Add green polygons for Polygon or MultiPolygon geometries
             for polygon in coords:
                 m.add_polygon(Polygon([(c[0], c[1]) for c in polygon], 'green', 3))
-
-    image = m.render(zoom=10)
-    return Image.fromarray(image)
+    
+    # Render the static map and return the Pillow Image object
+    return m.render(zoom=10)
 
 # ControlNet pipeline setup
 controlnet = ControlNetModel.from_pretrained("lllyasviel/control_v11p_sd15_inpaint", torch_dtype=torch.float16)