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