from datasets import load_dataset
import streamlit as st

# all levels, validation split (public)
ds_val = load_dataset(
    "gaia-benchmark/GAIA",
    "2023_all",
    split="validation",
    token=True,                 # use_auth_token=True on older installs also works
    trust_remote_code=True,     # needed because GAIA uses a loading script
)

ds_val.set_format(type="pandas")
df_val = ds_val[:]

temp_id = df_val.at[0, 'task_id']

st.title('Task_id 展示')
st.write('Task_id:', temp_id)