yoinked commited on
Commit
7d57a3e
1 Parent(s): c9e9b2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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
- output_text = ' '.join(text_items)
 
 
 
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='Chen " " Chen Chen "_"')
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')