Spaces:
Running
Running
kovacsvi
commited on
Commit
·
1f27218
1
Parent(s):
71aee23
debug prints
Browse files- interfaces/cap_minor.py +3 -0
interfaces/cap_minor.py
CHANGED
@@ -134,6 +134,9 @@ def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):
|
|
134 |
filtered_probs.items(), key=lambda item: item[1], reverse=True
|
135 |
)
|
136 |
}
|
|
|
|
|
|
|
137 |
|
138 |
# If eligible for minor prediction
|
139 |
if top_major_id in major_to_minor_map:
|
|
|
134 |
filtered_probs.items(), key=lambda item: item[1], reverse=True
|
135 |
)
|
136 |
}
|
137 |
+
|
138 |
+
print(top_major_id) # debug
|
139 |
+
print(major_to_minor_map) # debug
|
140 |
|
141 |
# If eligible for minor prediction
|
142 |
if top_major_id in major_to_minor_map:
|