import streamlit as st
import datasets
from datasets import load_dataset
import pandas as pd 


@st.cache
def load_data(dataset_name):
    dataset = load_dataset(dataset_name)
    return dataset
    
dataset = load_data("cornell_movie_dialog")   
st.write("loading dataset")
st.dataframe(dataset)
st.write("loaded")