Update app.py

app.py

@@ -1,17 +1,17 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the pre-trained model
-code_analyzer = pipeline("text-classification", model="huggingface/codebert-base-vulnerability-detection")
+# Load the pre-trained model for detecting insecure code
+code_analyzer = pipeline("text-classification", model="mrm8488/codebert-base-finetuned-detect-insecure-code")
 
 # Function to analyze code snippets
 def analyze_code(code_snippet):
     result = code_analyzer(code_snippet)
-    if result[0]["label"] == "VULNERABLE":
+    if result[0]["label"] == "LABEL_1":  # Assuming LABEL_1 means insecure
         return (
-            f"⚠️ Potential Issue Detected: {result[0]['label']} "
+            f"⚠️ Potential Issue Detected: Insecure Code "
             f"(Confidence: {result[0]['score']:.2f})\n"
-            "💡 Suggestion: Avoid using unsafe practices like 'eval'. Replace it with safer alternatives."
+            "💡 Suggestion: Avoid using unsafe practices."
         )
     else:
         return "✅ Code appears secure!"