Update app.py

app.py

@@ -406,7 +406,16 @@ def res_source():
         return "__Source__ Reset"
     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:
         pass_box = gr.Textbox()