Update app.py

app.py

@@ -8,18 +8,20 @@ model_name = "Salesforce/codegen-350M-mono"  # Fastest model for code review
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name).to("cpu")  # Force CPU mode
 
-# ✅ Function to review Python code with debug logs
 def review_code(code_snippet):
     print("✅ Received Code:", code_snippet)  # Debugging log
     
+    # ✅ Add a clear instruction to the model
+    prompt = f"### Instruction: Review and fix the Python code below.\n### Input Code:\n{code_snippet}\n### Fixed Code:\n"
+
     # Process input
-    inputs = tokenizer(code_snippet, return_tensors="pt").to("cpu")  # Move to CPU
+    inputs = tokenizer(prompt, return_tensors="pt").to("cpu")  # Move to CPU
     outputs = model.generate(
         **inputs,
-        max_length=50,  # ✅ Reduced token generation to prevent hallucinations
+        max_length=80,  # ✅ Increased length for full function review
         do_sample=False,
-        num_beams=3,
-        repetition_penalty=2.0  # ✅ Penalizes repetitive text generation
+        num_beams=5,  # ✅ Higher beams for better decision-making
+        repetition_penalty=1.8  # ✅ Lower penalty to avoid weird token removal
     )
 
     # Check if the model generated output
@@ -37,6 +39,7 @@ def review_code(code_snippet):
 
     return reviewed_code, temp_file_path  # ✅ Return reviewed code & file path
 
+
 # ✅ Handle user input and return reviewed code
 def check_code(input_code):
     reviewed_code, file_path = review_code(input_code)