import streamlit as st
from transformers import pipeline
from datasets import load_dataset
import pandas as pd
data_files = {'train': 'train.csv', 'test': 'test.csv'}

dataset = load_dataset('csv',data_files='train_input_smallnn.csv', split='train')
df = dataset.set_format(type='pandas')
st.table(df)