IdlecloudX commited on
Commit
6ba3022
·
verified ·
1 Parent(s): 7f24625

Update app.py

Browse files

fixed '[character'](KeyError: 'character')

Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -87,19 +87,17 @@ class Tagger:
87
  res["ratings"][tag_name] = float(outputs[idx])
88
  tag_categories_for_translation["ratings"].append(tag_name)
89
 
90
-
91
  for idx in self.categories["general"]:
92
  if outputs[idx] > gen_th:
93
  tag_name = self.tag_names[idx].replace("_", " ")
94
  res["general"][tag_name] = float(outputs[idx])
95
  tag_categories_for_translation["general"].append(tag_name)
96
 
97
-
98
  for idx in self.categories["character"]:
99
  if outputs[idx] > char_th:
100
  tag_name = self.tag_names[idx].replace("_", " ")
101
  res["characters"][tag_name] = float(outputs[idx])
102
- tag_categories_for_translation["character"].append(tag_name)
103
 
104
 
105
  res["general"] = dict(sorted(res["general"].items(), key=lambda kv: kv[1], reverse=True))
 
87
  res["ratings"][tag_name] = float(outputs[idx])
88
  tag_categories_for_translation["ratings"].append(tag_name)
89
 
 
90
  for idx in self.categories["general"]:
91
  if outputs[idx] > gen_th:
92
  tag_name = self.tag_names[idx].replace("_", " ")
93
  res["general"][tag_name] = float(outputs[idx])
94
  tag_categories_for_translation["general"].append(tag_name)
95
 
 
96
  for idx in self.categories["character"]:
97
  if outputs[idx] > char_th:
98
  tag_name = self.tag_names[idx].replace("_", " ")
99
  res["characters"][tag_name] = float(outputs[idx])
100
+ tag_categories_for_translation["characters"].append(tag_name)
101
 
102
 
103
  res["general"] = dict(sorted(res["general"].items(), key=lambda kv: kv[1], reverse=True))