Update app.py

app.py

@@ -112,16 +112,16 @@ class GPUSatelliteModelGenerator:
 
     @staticmethod
     @cp.fuse()  # Use CuPy's JIT compilation
-    def gpu_color_distance_hsv(pixel_hsv, reference_hsv, tolerance):
+    def gpu_color_distance_hsv(pixel_hsv, reference_hsv, hue_tolerance, sat_tolerance, val_tolerance):
         """Optimized HSV color distance calculation using CuPy's JIT"""
         h_diff = cp.minimum(cp.abs(pixel_hsv[0] - reference_hsv[0]),
                           360 - cp.abs(pixel_hsv[0] - reference_hsv[0]))
         s_diff = cp.abs(pixel_hsv[1] - reference_hsv[1])
         v_diff = cp.abs(pixel_hsv[2] - reference_hsv[2])
         
-        return (h_diff <= tolerance['hue']) & \
-               (s_diff <= tolerance['sat']) & \
-               (v_diff <= tolerance['val'])
+        return (h_diff <= hue_tolerance) & \
+               (s_diff <= sat_tolerance) & \
+               (v_diff <= val_tolerance)
 
     def generate_tree_vertices(self, tree_mask, base_vertices):
         """Generate randomized tree heights and positions"""
@@ -173,7 +173,9 @@ class GPUSatelliteModelGenerator:
                 masks[category] |= self.gpu_color_distance_hsv(
                     cp.stack([h, s, v]),
                     ref_hsv,
-                    tolerance
+                    tolerance['hue'],
+                    tolerance['sat'],
+                    tolerance['val']
                 )
         
         # Optimized terrain and building detection