Update app.py

app.py

@@ -1,6 +1,15 @@
 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(
@@ -8,6 +17,11 @@ classifier = pipeline(
     model="CIRCL/cwe-parent-vulnerability-classification-roberta-base",
     return_all_scores=True
 )
+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:
@@ -41,8 +55,6 @@ 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"]
     ]
 )