Update app.py

app.py

@@ -5,8 +5,5 @@ import pandas as pd
 data_files = {'train': 'train.csv', 'test': 'test.csv'}
 
 dataset = load_dataset('csv',data_files='train_input_smallnn.csv', split='train')
-dataset.set_format(type='pandas')
-
-df = dataset[:]
-
-st.table(df)
+df = dataset.set_format(type='pandas')
+st.table(df.head())