Update app.py

app.py

@@ -1,23 +1,13 @@
 import gradio as gr
 import json
 from transformers import pipeline
-import torch
-import random
-import numpy as np
-
-torch.manual_seed(42)
-random.seed(42)
-np.random.seed(42)
-
-torch.use_deterministic_algorithms(True)
 
 # Load Hugging Face model (text classification)
 classifier = pipeline(
     task="text-classification",
     model="CIRCL/cwe-parent-vulnerability-classification-roberta-base",
-    top_k=None
+    return_all_scores=True
 )
-classifier.model.eval()
 
 # Load child-to-parent mapping
 with open("child_to_parent_mapping.json", "r") as f:
@@ -29,9 +19,6 @@ 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 = {}
@@ -54,6 +41,8 @@ demo = gr.Interface(
         ["Fixed buffer overflow in input parsing"],
         ["SQL injection possible in login flow"],
         ["Improved input validation to prevent XSS"],
+        ["Added try/catch to avoid null pointer crash"],
+        ["Patched race condition in thread lock logic"]
     ]
 )