Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -149,7 +149,7 @@ def image_to_wd14_tags(image: Image.Image, model_name: str, threshold: float,
|
|
| 149 |
text_items.append(tag_outformat)
|
| 150 |
output_text = ', '.join(text_items)
|
| 151 |
|
| 152 |
-
return
|
| 153 |
|
| 154 |
|
| 155 |
if __name__ == '__main__':
|
|
@@ -176,6 +176,6 @@ if __name__ == '__main__':
|
|
| 176 |
gr_btn_submit.click(
|
| 177 |
image_to_wd14_tags,
|
| 178 |
inputs=[gr_input_image, gr_model, gr_threshold, gr_space, gr_escape, gr_confidence, gr_order],
|
| 179 |
-
outputs=[
|
| 180 |
)
|
| 181 |
demo.launch()
|
|
|
|
| 149 |
text_items.append(tag_outformat)
|
| 150 |
output_text = ', '.join(text_items)
|
| 151 |
|
| 152 |
+
return output_text, filtered_tags
|
| 153 |
|
| 154 |
|
| 155 |
if __name__ == '__main__':
|
|
|
|
| 176 |
gr_btn_submit.click(
|
| 177 |
image_to_wd14_tags,
|
| 178 |
inputs=[gr_input_image, gr_model, gr_threshold, gr_space, gr_escape, gr_confidence, gr_order],
|
| 179 |
+
outputs=[gr_output_text],
|
| 180 |
)
|
| 181 |
demo.launch()
|