Update app.py
Browse files
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}'
|