Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from datasets import load_dataset
 
 auth_token = os.environ.get("auth_token")
-visit_bench_all = load_dataset("mlfoundations/visit-bench", use_auth_token=auth_token)
+visit_bench_all = load_dataset("mlfoundations/VisIT-Bench", use_auth_token=auth_token)
 print('visit_bench_all')
 print(visit_bench_all)
 print('dataset keys:')
@@ -12,6 +12,8 @@ dataset_keys = list(visit_bench_all.keys())
 assert len(dataset_keys) == 1
 dataset_key = dataset_keys[0]
 visit_bench = visit_bench_all[dataset_key]
+print('first item:')
+print(visit_bench[0])
 
 df = visit_bench.to_pandas()
 print(f"Got {len(df)} items in dataframe")