Update app.py

app.py

@@ -32,7 +32,7 @@ def get_ckpt_names(token, radio_model_names, input_model):
 
     try:
         api = HfApi(token=token)
-        ckpt_files = [f for f in api.list_repo_files(repo_id=model_id) if f.endswith(".ckpt")]
+        ckpt_files = [f for f in api.list_repo_files(repo_id=model_id) if f.endswith(".ckpt") or f.endswith(".safetensors")]
         
         if not ckpt_files:
             return error_str("No checkpoint files found in the model repo."), gr.update(choices=[]), gr.update(visible=False)