Update app.py

app.py

@@ -445,7 +445,14 @@ def res_source():
     except Exception as e:
         return e
 
-
+f_ist = (api.list_repo_files(repo_id=f'{repo}/{name}', repo_type="dataset"))
+send_list =[]
+for i,ea in enumerate(f_ist):
+    if "balance/" in ea:
+        try:
+            send_list.append(ea.split("/",1)[1].split(".",1)[0])
+        except Exception:
+            pass
             
 with gr.Blocks() as bc:
     with gr.Row(visible=True) as invalid: