osbm commited on
Commit
fdf8211
·
1 Parent(s): e19621f
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -25,8 +25,11 @@ with st.form("login"):
25
 
26
 
27
  def get_list_of_images():
28
- fs = hfh.HfFileSystem(token=os.environ.get("hf_token"))
29
- return fs.ls("datasets/aifred-smart-life-coach/capstone-images", detail=False, refresh=True)
 
 
 
30
 
31
 
32
  def get_one_from_queue(voter: str):
 
25
 
26
 
27
  def get_list_of_images():
28
+ # fs = hfh.HfFileSystem(token=os.environ.get("hf_token"))
29
+ # return fs.ls("datasets/aifred-smart-life-coach/capstone-images", detail=False, refresh=True)
30
+ api = hfh.HfApi()
31
+ files = api.list_repo_tree(repo_id="aifred-smart-life-coach/capstone-images", repo_type="dataset")
32
+ return files
33
 
34
 
35
  def get_one_from_queue(voter: str):