FrostIce commited on
Commit
ffa6479
·
verified ·
1 Parent(s): 0477e21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 ratings, output_text, filtered_tags
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=[return,gr_output_text],
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()