Update app.py

app.py

@@ -15,14 +15,10 @@ torch.use_deterministic_algorithms(True)
 classifier = pipeline(
     task="text-classification",
     model="CIRCL/cwe-parent-vulnerability-classification-roberta-base",
-    return_all_scores=True
+    top_k=None
 )
 classifier.model.eval()
 
-threshold = 0.2
-filtered_results = [item for item in sorted_results if item["score"] >= threshold]
-
-
 # Load child-to-parent mapping
 with open("child_to_parent_mapping.json", "r") as f:
     child_to_parent = json.load(f)
@@ -33,6 +29,9 @@ def predict_cwe(commit_message: str):
     """
     results = classifier(commit_message)[0]
     sorted_results = sorted(results, key=lambda x: x["score"], reverse=True)
+        
+    threshold = 0.2
+    filtered_results = [item for item in sorted_results if item["score"] >= threshold]
 
     # Map predictions to parent CWE (if available)
     mapped_results = {}