Update app.py

app.py

@@ -30,7 +30,7 @@ def generate_code(prompt):
     with torch.no_grad():
         outputs = model.generate(
             **inputs,
-            max_new_tokens=512,
+            max_new_tokens=256,
             temperature=0.7
         )