fix: param name

app.py

@@ -10,7 +10,7 @@ client = InferenceClient(model="pszemraj/distilgpt2-magicprompt-SD")
 def respond(
     message,
     _: list[tuple[str, str]],
-    max_tokens: int,
+    max_new_tokens: int,
     temperature: float,
     top_p: float,
     top_k: int
@@ -19,7 +19,7 @@ def respond(
 
     for message in client.text_generation(
         message,
-        max_tokens=max_tokens,
+        max_new_tokens=max_new_tokens,
         stream=True,
         temperature=temperature,
         top_p=top_p,
@@ -37,7 +37,7 @@ For information on how to customize the ChatInterface, peruse the gradio docs: h
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
-        gr.Slider(minimum=8, maximum=128, value=64, step=8, label="Max tokens"),
+        gr.Slider(minimum=8, maximum=128, value=64, step=8, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(
             minimum=0.1,