Update app.py

app.py

@@ -39,13 +39,17 @@ def load_sfts(repo_1_id, repo_2_id):
         sfts_available_files_1 = fs.glob(f"{repo_1_id}/*.safetensors")
         sfts_available_files_1 = get_files(sfts_available_files_1)
     except:
-        sfts_available_files_1 = "NO SAFETENSORS FILE"
-        
+        sfts_available_files_1 = ["NO SAFETENSORS FILE"]  
         pass
     
     print(f"sfts 1: {sfts_available_files_1}")
-    sfts_available_files_2 = fs.glob(f"{repo_2_id}/*.safetensors")
-    sfts_available_files_2 = get_files(sfts_available_files_2)
+
+    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 = ["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)