Retrieve list of configs

app.py

@@ -1,4 +1,4 @@
-from datasets import load_dataset, IterableDataset
+from datasets import load_dataset, get_dataset_config_names
 from functools import partial
 from pandas import DataFrame
 import gradio as gr 
@@ -7,6 +7,7 @@ import tqdm
 import json
 import os
 
+DATASET = "satellogic/EarthView"
 DEBUG = False
 
 sets = {
@@ -116,20 +117,20 @@ def new_state():
 with gr.Blocks(title="Dataset Explorer", fill_height = True) as demo:
     state = new_state()
 
-    gr.Markdown("# Viewer for [satellogic/EarthView](https://huggingface.co/datasets/satellogic/EarthView) Dataset")
+    gr.Markdown(f"# Viewer for [{DATASET}](https://huggingface.co/datasets/satellogic/EarthView) Dataset")
     batch_size = gr.Number(10, label = "Batch Size", render=False)
     shard = gr.Slider(label="Shard", minimum=0, maximum=10000, step=1, render=False)
     table = gr.DataFrame(render = False)
     # headers=["Index","TimeStamp","Bounds","CRS"], 
 
     gallery = gr.Gallery(
-        label="satellogic/EarthView",
+        label=DATASET,
         interactive=False,
         columns=5, rows=2, render=False)
 
     with gr.Row():
-        dataset = gr.Textbox(label="Dataset", value="satellogic/EarthView")
-        config = gr.Dropdown(choices=["satellogic", "sentinel_1", "neon"], label="Subset", value="satellogic", )
+        dataset = gr.Textbox(label="Dataset", value=DATASET, interactive=False)
+        config = gr.Dropdown(choices=get_dataset_config_names(DATASET), label="Config", value="satellogic", )
         split = gr.Textbox(label="Split", value="train")
         initial_shard = gr.Number(label = "Initial shard", value=0, info="-1 for whole dataset")