npc

interfaces/cap_minor.py

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