fix?
This commit is contained in:
parent
4b8202f493
commit
8449d73675
1 changed files with 6 additions and 2 deletions
8
main.py
8
main.py
|
@ -106,8 +106,12 @@ class SDInterrogator(Interrogator):
|
|||
log.info("got %d", resp.status)
|
||||
assert resp.status == 200
|
||||
data = await resp.json()
|
||||
tags_with_scores = data["caption"]
|
||||
tags = list(tags_with_scores.keys())
|
||||
tags_with_weights = data["caption"]
|
||||
|
||||
tags = []
|
||||
|
||||
for tag, weight in tags_with_weights.items():
|
||||
tags.append(tag)
|
||||
|
||||
upstream_tags = [tag.replace(" ", "_") for tag in tags]
|
||||
return " ".join(upstream_tags)
|
||||
|
|
Loading…
Reference in a new issue