Update app.py

app.py

@@ -145,14 +145,21 @@ _CLASSES = [
 # test
 # l, im, f = main(*slider_defaults)
 
-#load both datasets in memory? --> easier retrieval
+#load both datasets in memory? --> easier retrieval afterwards with seed index based on pressing button
+DATASETS = {
+    'rvl_cdip': load_dataset(
+        "bdpc/rvl_cdip_mp",
+        split="test"), 
+    'rvl_cdip_N': load_dataset(
+        "bdpc/rvl_cdip_mp",
+        split="test")
+}
 
 meta_cats = {'dataset': ['rvl_cdip', 'rvl_cdip_N'],
              'label': _CLASSES
             }
 sliders = [gr.Dropdown(choices=choices, value=choices[-1], label=label) for label, choices in meta_cats.items()]
-slider_defaults = [None, "visual_checkbox", None, None, None]  # [slider.value for slider in sliders]
-
+slider_defaults = [slider.value for slider in sliders]
 
 outputs = [
     gr.Textbox(label="label"),