Update app.py

app.py

@@ -1,12 +1,13 @@
 import gradio as gr
 from datasets import load_dataset
 import huggingface_hub as hh
+from huggingface_hub import HfApi
 import os
 
 hh.login(token=os.environ['token'])
 
-# If the dataset is gated/private, make sure you have run huggingface-cli login
-dataset = load_dataset("AllenYkl/baikejiankang")
+api = HfApi()
+file_list = api.list_repo_files(repo_id=os.environ['repo_id'], repo_type="space")
 
 print(dataset)