num2skip, num2embed

app.py

@@ -44,13 +44,15 @@ def download(
     ds_name,
     ds_config,
     ds_split,
+    num2skip,
+    num2embed,
     progress=gr.Progress(),
 ):
     if progress is not None:
         progress(0.5, "Loading dataset...")
     
     if ds_name == "wikipedia":
-        ds = download_wikipedia(ds_name, ds_config)
+        ds = download_wikipedia(ds_name, ds_config, num2skip, num2embed)
     else:
         ds = load_hf_dataset(ds_name, ds_config, ds_split)
 
@@ -194,6 +196,8 @@ with gr.Blocks(title="Bulk embeddings") as demo:
             ds_name,
             ds_config,
             ds_split,
+            num2skip,
+            num2embed,
         ],
         outputs=last,
     )