app.py

import streamlit as st
from transformers import pipeline
from datasets import load_dataset

data_files = {"train": "train_input_smallnn.csv", "val": "val_input_smallnn.csv"}
dataset = load_dataset('puqi/test',data_files=data_files)

st.table(dataset)