Spaces:
Runtime error
Runtime error
Return missing tags
Browse files
app.py
CHANGED
@@ -94,6 +94,8 @@ def run(
|
|
94 |
|
95 |
tags = general_tags + hair_color_tags + hair_style_tags + \
|
96 |
eye_color_tags + image_color_tags + other_tags + additional_tags
|
|
|
|
|
97 |
tag_indices = [
|
98 |
deepdanbooru_tag_dict[tag] for tag in tags
|
99 |
if tag in deepdanbooru_tag_dict
|
@@ -128,7 +130,7 @@ def run(
|
|
128 |
ncols * image_size, 3)
|
129 |
seeds = np.asarray(seeds).reshape(nrows, ncols)
|
130 |
|
131 |
-
return len(image_indices), res, seeds
|
132 |
|
133 |
|
134 |
def main():
|
@@ -239,6 +241,7 @@ def main():
|
|
239 |
gr.outputs.Textbox(type='number', label='Number of Found Images'),
|
240 |
gr.outputs.Image(type='numpy', label='Output'),
|
241 |
gr.outputs.Dataframe(type='numpy', label='Seed'),
|
|
|
242 |
],
|
243 |
title=TITLE,
|
244 |
description=DESCRIPTION,
|
|
|
94 |
|
95 |
tags = general_tags + hair_color_tags + hair_style_tags + \
|
96 |
eye_color_tags + image_color_tags + other_tags + additional_tags
|
97 |
+
missing_tags = [tag for tag in tags if tag not in deepdanbooru_tag_dict]
|
98 |
+
|
99 |
tag_indices = [
|
100 |
deepdanbooru_tag_dict[tag] for tag in tags
|
101 |
if tag in deepdanbooru_tag_dict
|
|
|
130 |
ncols * image_size, 3)
|
131 |
seeds = np.asarray(seeds).reshape(nrows, ncols)
|
132 |
|
133 |
+
return len(image_indices), res, seeds, ','.join(missing_tags)
|
134 |
|
135 |
|
136 |
def main():
|
|
|
241 |
gr.outputs.Textbox(type='number', label='Number of Found Images'),
|
242 |
gr.outputs.Image(type='numpy', label='Output'),
|
243 |
gr.outputs.Dataframe(type='numpy', label='Seed'),
|
244 |
+
gr.outputs.Textbox(type='str', label='Missing Tags'),
|
245 |
],
|
246 |
title=TITLE,
|
247 |
description=DESCRIPTION,
|