Update app.py

app.py

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