fix quantile

app.py

@@ -3269,9 +3269,13 @@ with demo:
                     
                     def to_mask(heatmap, threshold):
                         heatmap = 1 / (heatmap + 1e-6)
-                        heatmap = heatmap.reshape(len(ncut_images), h, w)
-                        vmin, vmax = heatmap.quantile(0.01), heatmap.quantile(0.99)
+                        if heatmap.shape[0] > 10000:
+                            random_idx = np.random.choice(heatmap.shape[0], 10000, replace=False)
+                            vmin, vmax = heatmap[random_idx].quantile(0.01), heatmap[random_idx].quantile(0.99)
+                        else:
+                            vmin, vmax = heatmap.quantile(0.01), heatmap.quantile(0.99)
                         heatmap = (heatmap - vmin) / (vmax - vmin)
+                        heatmap = heatmap.reshape(len(ncut_images), h, w)
                         mask = heatmap > threshold
                         return mask