max_new_tokens

app.py

@@ -14,7 +14,7 @@ public class HelloWorld {
 """
 
 inputs = tokenizer(code_snippet, return_tensors="pt")
-outputs = model.generate(**inputs)
+outputs = model.generate(**inputs, max_new_tokens=100)
 review = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
 print("Code Review:", review)