Update app.py

app.py

@@ -3,7 +3,10 @@ import gradio as gr
 from datasets import load_dataset
 
 auth_token = os.environ.get("auth_token")
-visit_bench = load_dataset("mlfoundations/visit-bench", use_auth_token=auth_token)['test']
+visit_bench_all = load_dataset("mlfoundations/visit-bench", use_auth_token=auth_token)
+print('dataset keys:')
+print(visit_bench_all.keys())
+visit_bench = visit_bench_all['test']
 
 df = visit_bench.to_pandas()
 print(f"Got {len(df)} items in dataframe")