Update app.py

app.py

@@ -283,7 +283,7 @@ PROMPTS:
         inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
         outputs = model.generate(
             **inputs,
-            max_new_tokens=min(ax_tokens,512),
+            max_new_tokens=min(max_tokens,512),
             temperature=0.3,
             do_sample=False,
             pad_token_id=tokenizer.eos_token_id