Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from transformers import pipeline
 from datasets import load_dataset
 
-data_files = {"train": "train_input_smallnn.csv", "val": "val_input_smallnn.csv"}
-dataset = load_dataset('puqi/test',data_files=data_files)
 
-st.table(dataset['val'])
+dataset = load_dataset('puqi/test',data_files="train_input_smallnn.csv")
+
+st.table(dataset)