Symbiomatrix commited on
Commit
3ebe424
·
verified ·
1 Parent(s): 8abd332

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -117,7 +117,7 @@ def get_model_meta(model):
117
  }
118
  return (dfiles, dimages)
119
 
120
- def get_files_by_username(username, api_key=None, nsfw=None, hidden=None):
121
  url = f"https://civitai.com/api/v1/models?username={username}&limit=100"
122
  output = {}
123
  images = {}
@@ -126,6 +126,8 @@ def get_files_by_username(username, api_key=None, nsfw=None, hidden=None):
126
  url = url + f"&nsfw={BOOLPARM[nsfw]}"
127
  if hidden is not None:
128
  url = url + f"&hidden={BOOLPARM[hidden]}"
 
 
129
  gr.Info(f"SBM url: {url}")
130
  if api_key:
131
  headers['Authorization'] = f'Bearer {api_key}'
 
117
  }
118
  return (dfiles, dimages)
119
 
120
+ def get_files_by_username(username, api_key=None, nsfw=None, hidden=None, favourites=None):
121
  url = f"https://civitai.com/api/v1/models?username={username}&limit=100"
122
  output = {}
123
  images = {}
 
126
  url = url + f"&nsfw={BOOLPARM[nsfw]}"
127
  if hidden is not None:
128
  url = url + f"&hidden={BOOLPARM[hidden]}"
129
+ if favourites: # fav=false does nothing.
130
+ url = url + f"&favorites={BOOLPARM[favourites]}"
131
  gr.Info(f"SBM url: {url}")
132
  if api_key:
133
  headers['Authorization'] = f'Bearer {api_key}'