Update app.py

app.py

@@ -35,23 +35,21 @@ def get_files(file_paths):
 def load_sfts(repo_1_id, repo_2_id):
       
     # List all ".safetensors" files in repos
-    try:
-        sfts_available_files_1 = fs.glob(f"{repo_1_id}/*.safetensors")
-        sfts_available_files_1 = get_files(sfts_available_files_1)
-        if sfts_available_files_1 == []:
-            sfts_available_files_1 = ["NO SAFETENSORS FILE"]
-    except:
-        sfts_available_files_1 = ["NO SAFETENSORS FILE"]  
-        pass
+    
+    sfts_available_files_1 = fs.glob(f"{repo_1_id}/*.safetensors")
+    sfts_available_files_1 = get_files(sfts_available_files_1)
+    
+    if sfts_available_files_1 == []:
+        sfts_available_files_1 = ["NO SAFETENSORS FILE"]
     
     print(f"sfts 1: {sfts_available_files_1}")
 
-    try:
-        sfts_available_files_2 = fs.glob(f"{repo_2_id}/*.safetensors")
-        sfts_available_files_2 = get_files(sfts_available_files_2)
-    except:
+    
+    sfts_available_files_2 = fs.glob(f"{repo_2_id}/*.safetensors")
+    sfts_available_files_2 = get_files(sfts_available_files_2)
+    
+    if sfts_available_files_2 == []:
         sfts_available_files_2 = ["NO SAFETENSORS FILE"]
-        pass
     
     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)