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


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