npc

interfaces/cap_minor.py

@@ -124,7 +124,7 @@ def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):
 
     # Default: show major topic predictions
     if int(top_major_id) == 999:
-        filtered_probs = {-1:1}
+        filtered_probs = {21:1}
     else:
         filtered_probs = {
             i: float(major_probs_np[i]) for i in np.argsort(major_probs_np)[::-1]