kovacsvi commited on
Commit
1f27218
·
1 Parent(s): 71aee23

debug prints

Browse files
Files changed (1) hide show
  1. 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: