fffiloni commited on
Commit
94a2387
1 Parent(s): 1d76b5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -35,7 +35,8 @@ def get_files(file_paths):
35
  def load_sfts(repo_1_id, repo_2_id):
36
  your_username = api.whoami()["name"]
37
  my_models = api.list_models(author=your_username, filter=["diffusers", "stable-diffusion-xl", 'lora'])
38
- print(my_models)
 
39
  # List all ".safetensors" files in repos
40
  sfts_available_files_1 = fs.glob(f"{repo_1_id}/*.safetensors")
41
  sfts_available_files_1 = get_files(sfts_available_files_1)
 
35
  def load_sfts(repo_1_id, repo_2_id):
36
  your_username = api.whoami()["name"]
37
  my_models = api.list_models(author=your_username, filter=["diffusers", "stable-diffusion-xl", 'lora'])
38
+ model_names = [item['Model Name'] for item in my_models]
39
+ print(model_names)
40
  # List all ".safetensors" files in repos
41
  sfts_available_files_1 = fs.glob(f"{repo_1_id}/*.safetensors")
42
  sfts_available_files_1 = get_files(sfts_available_files_1)