Kohaku-Blueleaf commited on
Commit
36f7056
1 Parent(s): 25e6739

change some default values

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -202,7 +202,7 @@ def numpy2pytorch(imgs):
202
  @torch.inference_mode()
203
  def interrogator_process(x):
204
  img = Image.fromarray(x)
205
- rating, features, chars = get_wd14_tags(img, general_threshold=0.25, no_underline=True)
206
  result = ""
207
  for char in chars:
208
  result += char
 
202
  @torch.inference_mode()
203
  def interrogator_process(x):
204
  img = Image.fromarray(x)
205
+ rating, features, chars = get_wd14_tags(img, general_threshold=0.3, no_underline=True)
206
  result = ""
207
  for char in chars:
208
  result += char