Update app.py

app.py

@@ -3,6 +3,6 @@ from transformers import pipeline
 from datasets import load_dataset
 
 
-dataset = load_dataset(path='puqi/test',data_files="train_input_smallnn.csv")
+dataset = load_dataset('puqi/test/', split='train')
 
 st.table(dataset)