yuzhounie commited on
Commit
8f11315
·
1 Parent(s): 071ed23

Update model name for nopolicy in app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -43,6 +43,8 @@ def analyze_code(code: str, language: str, model: str) -> str:
43
  model_used = result.get("result", {}).get("model", model)
44
  explanation = result.get("result", {}).get("response", "")
45
 
 
 
46
  if status == "yes":
47
  output = f"⚠️ **Vulnerability Detected**\n\n"
48
  output += f"**CWE Type:** {cwe_type}\n"
 
43
  model_used = result.get("result", {}).get("model", model)
44
  explanation = result.get("result", {}).get("response", "")
45
 
46
+ if "nopolicy" in model_used:
47
+ model_used = "VirtueGuard Code"
48
  if status == "yes":
49
  output = f"⚠️ **Vulnerability Detected**\n\n"
50
  output += f"**CWE Type:** {cwe_type}\n"