Update app.py

app.py

@@ -30,7 +30,7 @@ def generate_code(prompt):
     with torch.no_grad():
         generated_ids = model.generate(
             **model_inputs,
-            max_new_tokens=1024
+            max_new_tokens=384
         )
     generated_ids = [
         output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)