Update app.py

app.py

@@ -14,7 +14,13 @@ def review_code(code_snippet):
     
     # Process input
     inputs = tokenizer(code_snippet, return_tensors="pt").to("cpu")  # Move to CPU
-    outputs = model.generate(**inputs, max_length=80, do_sample=False, num_beams=3)
+    outputs = model.generate(
+        **inputs,
+        max_length=50,  # ✅ Reduced token generation to prevent hallucinations
+        do_sample=False,
+        num_beams=3,
+        repetition_penalty=2.0  # ✅ Penalizes repetitive text generation
+    )
 
     # Check if the model generated output
     if outputs is None: