Update app.py

app.py

@@ -30,7 +30,7 @@ def generate_code(prompt):
     with torch.no_grad():
         generated_ids = model.generate(
             **model_inputs,
-            max_length = 1024,
+            max_new_tokens = 1024,
             min_new_tokens = 256,
             #low_memory = True,
             do_sample = True,