Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -166,7 +166,10 @@ def image_to_wd14_tags(image: Image.Image, model_name: str, threshold: float,
|
|
| 166 |
if include_ranks:
|
| 167 |
tag_outformat = f"({tag_outformat}:{score:.3f})"
|
| 168 |
text_items.append(tag_outformat)
|
| 169 |
-
|
|
|
|
|
|
|
|
|
|
| 170 |
|
| 171 |
return ratings, output_text, filtered_tags
|
| 172 |
|
|
@@ -180,7 +183,7 @@ if __name__ == '__main__':
|
|
| 180 |
gr_model = gr.Radio(list(WAIFU_MODELS.keys()), value='chen-moat2', label='Chen')
|
| 181 |
gr_threshold = gr.Slider(0.0, 1.0, 0.5, label='Chen Chen Chen Chen Chen')
|
| 182 |
with gr.Row():
|
| 183 |
-
gr_space = gr.Checkbox(value=True, label='
|
| 184 |
gr_escape = gr.Checkbox(value=True, label='Chen Text Escape')
|
| 185 |
|
| 186 |
gr_btn_submit = gr.Button(value='橙', variant='primary')
|
|
|
|
| 166 |
if include_ranks:
|
| 167 |
tag_outformat = f"({tag_outformat}:{score:.3f})"
|
| 168 |
text_items.append(tag_outformat)
|
| 169 |
+
if use_spaces:
|
| 170 |
+
output_text = ' '.join(text_items)
|
| 171 |
+
else:
|
| 172 |
+
output_text = ', '.join(text_items)
|
| 173 |
|
| 174 |
return ratings, output_text, filtered_tags
|
| 175 |
|
|
|
|
| 183 |
gr_model = gr.Radio(list(WAIFU_MODELS.keys()), value='chen-moat2', label='Chen')
|
| 184 |
gr_threshold = gr.Slider(0.0, 1.0, 0.5, label='Chen Chen Chen Chen Chen')
|
| 185 |
with gr.Row():
|
| 186 |
+
gr_space = gr.Checkbox(value=True, label='Use DashSpace')
|
| 187 |
gr_escape = gr.Checkbox(value=True, label='Chen Text Escape')
|
| 188 |
|
| 189 |
gr_btn_submit = gr.Button(value='橙', variant='primary')
|