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