fix: generate

app.py

@@ -245,25 +245,20 @@ with gr.Blocks(
             )
 
             # Event handlers for button actions to process and clear text
+            def format_with_style(text, style):
+                return generate(
+                    text,
+                    temperature=0.3,
+                    max_tokens=1024,
+                    top_k=40,
+                    repetition_penalty=1.1,
+                    top_p=0.95,
+                    prompt_style=STYLE_PROMPTS[style]
+                )
+
             generate_btn.click(
-                fn=lambda input_text, temp, max_t, top_k, rep_pen, top_p, style: generate(
-                    input_text,
-                    temp,
-                    max_t,
-                    top_k,
-                    rep_pen,
-                    top_p,
-                    STYLE_PROMPTS[style]
-                ),
-                inputs=[
-                    input_text,
-                    gr.Number(value=0.3, visible=False),  # temperature
-                    gr.Number(value=1024, visible=False),  # max_tokens
-                    gr.Number(value=40, visible=False),    # top_k
-                    gr.Number(value=1.1, visible=False),   # repetition_penalty
-                    gr.Number(value=0.95, visible=False),  # top_p
-                    style_dropdown,
-                ],
+                fn=format_with_style,
+                inputs=[input_text, style_dropdown],
                 outputs=output_text,
             )