Update app.py

app.py

@@ -1,10 +1,8 @@
 import gradio as gr
 from transformers import pipeline
 
-# Replace this with your actual Hugging Face model repo (trained using classify.py)
 MODEL_NAME = "CIRCL/cwe-vulnerability-classification-codebert-base"
 
-# Load the text classification pipeline
 classifier = pipeline("text-classification", model=MODEL_NAME, return_all_scores=True)
 
 def classify_cwe(text):
@@ -13,7 +11,6 @@ def classify_cwe(text):
     sorted_results = sorted(results, key=lambda x: x["score"], reverse=True)
     return {res["label"]: round(res["score"], 4) for res in sorted_results[:5]}
 
-# Create the Gradio interface
 interface = gr.Interface(
     fn=classify_cwe,
     inputs=gr.Textbox(lines=5, placeholder="Enter vulnerability description..."),