Update app.py

app.py

@@ -33,9 +33,9 @@ def get_files(file_paths):
 
 def load_sfts(repo_1_id, repo_2_id):
     # List all ".safetensors" files in repos
-    sfts_available_files_1 = fs.ls(f"{repo_1_id}/**.safetensors", detail=False)
+    sfts_available_files_1 = fs.glob(f"{repo_1_id}/*.safetensors")
     sfts_available_files_1 = get_files(sfts_available_files_1)
-    sfts_available_files_2 = fs.ls(f"{repo_2_id}/**.safetensors", detail=False)
+    sfts_available_files_2 = fs.glob(f"{repo_2_id}/*.safetensors")
     sfts_available_files_2 = get_files(sfts_available_files_2)
     return gr.update(choices=sfts_available_files_1, value=sfts_available_files_1[0], visible=True), gr.update(choices=sfts_available_files_2, value=sfts_available_files_2[0], visible=True)