utils/load_dataset.py

import streamlit as st
import datasets as ds


@st.cache
def load_datasets():
    return ds.list_datasets(with_community_datasets=False)