ds is used afterall

app.py

@@ -24,8 +24,8 @@ sets = {
 }
 
 def open_dataset(dataset, set_name, split, batch_size, shard = -1):
-    # I should really move dsi to a gr.State
-    global dsi
+    # I should really move ds.config_name and dsi to a gr.State()
+    global dsi, ds
 
     if shard == -1:
         data_files = None
@@ -58,7 +58,7 @@ def open_dataset(dataset, set_name, split, batch_size, shard = -1):
     )
     
 def get_images(batch_size):
-    global dsi
+    global dsi, ds
         
     items = []
     metadatas = []