Update app.py

app.py

@@ -16,7 +16,7 @@ pipeline = transformers.pipeline(
 )
 
 # Define the function to generate text
-def generate_text(input_text, max_new_tokens=128, temperature=1.0, top_k=50, top_p=0.95):
+def generate_text(input_text, max_new_tokens, temperature, top_k, top_p):
     output = pipeline(
         input_text,
         max_new_tokens=max_new_tokens,
@@ -30,11 +30,11 @@ def generate_text(input_text, max_new_tokens=128, temperature=1.0, top_k=50, top
 iface = gr.Interface(
     fn=generate_text,
     inputs=[
-        gr.inputs.Textbox(lines=2, placeholder="Enter your text here...", label="Input Text"),
-        gr.inputs.Slider(minimum=1, maximum=200, default=128, step=1, label="Max New Tokens"),
-        gr.inputs.Slider(minimum=0.1, maximum=2.0, default=1.0, step=0.1, label="Temperature"),
-        gr.inputs.Slider(minimum=1, maximum=100, default=50, step=1, label="Top-k"),
-        gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.95, step=0.01, label="Top-p")
+        gr.Textbox(lines=2, placeholder="Enter your text here...", label="Input Text"),
+        gr.Slider(minimum=1, maximum=200, value=128, step=1, label="Max New Tokens"),
+        gr.Slider(minimum=0.1, maximum=2.0, value=1.0, step=0.1, label="Temperature"),
+        gr.Slider(minimum=1, maximum=100, value=50, step=1, label="Top-k"),
+        gr.Slider(minimum=0.0, maximum=1.0, value=0.95, step=0.01, label="Top-p")
     ],
     outputs="text",
     title="Text Generation Interface",