Update app.py

app.py

@@ -271,6 +271,8 @@ class GPUSatelliteModelGenerator:
     def generate_mesh_gpu(self, height_map, texture_img):
         """Generate optimized 3D mesh with tree geometry"""
         height_map_gpu = cp.asarray(height_map)
+        # Convert texture_img to CuPy array
+        texture_img_gpu = cp.asarray(texture_img)
         height, width = height_map.shape
         
         # Generate base vertices
@@ -278,7 +280,7 @@ class GPUSatelliteModelGenerator:
         vertices = cp.stack([x, height_map_gpu * self.building_height, z], axis=-1)
         
         # Detect tree areas and generate tree geometry
-        tree_mask = cp.all(texture_img == self.colors['dark_green'], axis=2)
+        tree_mask = cp.all(texture_img_gpu == self.colors['dark_green'], axis=2)
         vertices = self.generate_tree_vertices(tree_mask, vertices.reshape(-1, 3))
         
         # Normalize coordinates
@@ -293,6 +295,7 @@ class GPUSatelliteModelGenerator:
         
         # Create textured mesh
         return self.create_textured_mesh(vertices, faces, uvs, texture_img)
+        
 
     @staticmethod
     def generate_faces_gpu(height, width):