This commit is contained in:
Luna 2024-08-04 16:12:34 -03:00
parent 4b8202f493
commit 8449d73675

View file

@ -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)