Increase max rows

app.py

@@ -25,11 +25,10 @@ import gradio as gr
 
 """
 TODOs:
-- Try with more rows
-- Add TextGenerationLayer
-- Try with more rows
-- Export and serve an interactive HTML plot?
+- Improve representation layer (Try with llamacpp or TextGeneration)
 - Make it run on Zero GPU
+- Try with more rows (Current: 50_000/10_000 -> Minimal Targett: 1_000_000/20_000)
+- Export interactive plots and serve their HTML content (It doesn't work with gr.HTML)
 """
 
 load_dotenv()
@@ -46,8 +45,8 @@ logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
 )
 
-MAX_ROWS = 10_000
-CHUNK_SIZE = 2_000
+MAX_ROWS = 50_000
+CHUNK_SIZE = 10_000
 
 
 session = requests.Session()
@@ -250,17 +249,18 @@ def generate_topics(dataset, config, split, column, nested_column, plot_type):
                 title=dataset,
                 width=800,
                 height=700,
-                # arrowprops={
-                #     "arrowstyle": "wedge,tail_width=0.5",
-                #     "connectionstyle": "arc3,rad=0.05",
-                #     "linewidth": 0,
-                #     "fc": "#33333377",
-                # },
-                label_wrap_width=12,
-                label_over_points=True,
-                dynamic_label_size=True,
-                max_font_size=36,
-                min_font_size=4,
+                arrowprops={
+                    "arrowstyle": "wedge,tail_width=0.5",
+                    "connectionstyle": "arc3,rad=0.05",
+                    "linewidth": 0,
+                    "fc": "#33333377",
+                },
+                dynamic_label_size=False,
+                # label_wrap_width=12,
+                # label_over_points=True,
+                # dynamic_label_size=True,
+                # max_font_size=36,
+                # min_font_size=4,
             )
             if plot_type == "DataMapPlot"
             else base_model.visualize_documents(