Update README.md

README.md

@@ -17,8 +17,12 @@ tokenizer = AutoTokenizer.from_pretrained("Mr-Vicky-01/TP-FP")
 
 start = time.time()
 
-question = "give me a scan result"
-question = re.sub(r'[,?.]', '', question)
+vuln = 'String password = "password123";'
+vuln_desc = " Hardcoded credentials were found. This could allow an attacker to access sensitive resources. Replace hardcoded passwords with environment variables or a secure vault."
+scanner = "docker"
+
+question = f"""Vulnerability: {vuln} , Vulnerability_Description {vuln_desc} , Scanner: {scanner}"""
+question = re.sub(r"[,?.'\"']", '', question)
 inputs = tokenizer(question, return_tensors="pt").to(device)
 with torch.no_grad():
     logits = model(**inputs).logits