Symbiomatrix commited on
Commit
70c3933
·
verified ·
1 Parent(s): 46ab99a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -100,7 +100,7 @@ def get_model_meta(model):
100
  }
101
  return (dfiles, dimages)
102
 
103
- def get_files_by_username(username, api_key=None, nsfw=None, hidden=None):
104
  url = f"https://civitai.com/api/v1/models?username={username}&limit=100"
105
  output = {}
106
  images = {}
@@ -109,6 +109,8 @@ def get_files_by_username(username, api_key=None, nsfw=None, hidden=None):
109
  url = url + f"&nsfw={BOOLPARM[nsfw]}"
110
  if hidden is not None:
111
  url = url + f"&hidden={BOOLPARM[hidden]}"
 
 
112
  gr.Info(f"SBM url: {url}")
113
  if api_key:
114
  headers['Authorization'] = f'Bearer {api_key}'
 
100
  }
101
  return (dfiles, dimages)
102
 
103
+ def get_files_by_username(username, api_key=None, nsfw=None, hidden=None, favourites=None):
104
  url = f"https://civitai.com/api/v1/models?username={username}&limit=100"
105
  output = {}
106
  images = {}
 
109
  url = url + f"&nsfw={BOOLPARM[nsfw]}"
110
  if hidden is not None:
111
  url = url + f"&hidden={BOOLPARM[hidden]}"
112
+ if favourites: # fav=false does nothing.
113
+ url = url + f"&favorites={BOOLPARM[favourites]}"
114
  gr.Info(f"SBM url: {url}")
115
  if api_key:
116
  headers['Authorization'] = f'Bearer {api_key}'