Spaces:
Running
Running
kovacsvi
commited on
Commit
·
0f6a4f6
1
Parent(s):
882055b
typo
Browse files
interfaces/cap_minor_media.py
CHANGED
@@ -179,7 +179,7 @@ def predict_flat(text, model_id, tokenizer_id, HF_TOKEN=None):
|
|
179 |
# Minor topic
|
180 |
major_code = code // 100
|
181 |
major_label = CAP_MEDIA_LABEL_NAMES[major_code]
|
182 |
-
minor_label =
|
183 |
display = f"[{major_code}] {major_label} [{code}] {minor_label}"
|
184 |
|
185 |
output_pred[display] = prob
|
|
|
179 |
# Minor topic
|
180 |
major_code = code // 100
|
181 |
major_label = CAP_MEDIA_LABEL_NAMES[major_code]
|
182 |
+
minor_label = CAP_MIN_LABEL_NAMES[code]
|
183 |
display = f"[{major_code}] {major_label} [{code}] {minor_label}"
|
184 |
|
185 |
output_pred[display] = prob
|