Update app.py

app.py

@@ -32,7 +32,7 @@ def convert_to_2d_geometry(geom):
   if geom is None:
     return None
   elif geom.has_z:
-    return transform(lambda x, y, z: (x, y), shape)
+    return transform(lambda x, y, z: (x, y), geom)
   else:
     return geom