Update app.py

app.py

@@ -54,19 +54,8 @@ def get_completion(query, model, tokenizer):
 
 @spaces.GPU()
 def code_review(code_to_analyze):
-    two_shot_prompt = f"""First, understand the given code:
-Analyze the purpose, functionality, input sources, output destinations, and logical flow of the code. Identify any security-sensitive operations.
-
-Now, review the following code:
-
-{code_to_analyze}
-
-Provide a detailed review including:
-1. Potential security issues
-2. Potential logic vulnerabilities
-3. Suggestions for improvement
-
-Start each section with its number and title."""
+    two_shot_prompt = f"""find all vulnerabilities which in the code 
+{code_to_analyze} """
 
     full_response = get_completion(two_shot_prompt, model, tokenizer)