Update app.py

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
 from transformers import pipeline
 from datasets import load_dataset
+data_files = {'train': 'train.csv', 'test': 'test.csv'}
 
-
-dataset = load_dataset(path='puqi/test',data_files='train.csv')
+dataset = load_dataset(path='puqi/test',data_files=data_files)
 
 st.table(dataset)