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.head()) |