Update app.py

app.py

@@ -68,7 +68,7 @@ def chat_fn(message, history):
         with torch.no_grad():
             output_ids = model.generate(
                 input_ids,
-                max_new_tokens=512,
+                max_tokens=512,
                 temperature=0.8,
                 top_k=50,
             )