Update app.py

app.py

@@ -13,19 +13,17 @@ def generate_text(input_text, temperature=0.9, max_new_tokens=50, top_p=0.95, to
 
     
 
-# Define your Gradio interface
-iface = gr.Interface(
-    fn=generate_text,
-    inputs=[
-        gr.inputs.Textbox(lines=2, label="Input Text"),
-        gr.inputs.Slider(minimum=0, maximum=1, step=0.01, default=0.9, label="Temperature"),
-        gr.inputs.Slider(minimum=1, maximum=100, step=1, default=50, label="Max New Tokens"),
-        gr.inputs.Slider(minimum=0, maximum=1, step=0.01, default=0.95, label="Top-p"),
-        gr.inputs.Slider(minimum=0, maximum=100, step=1, default=50, label="Top-k")
-    ],
-    outputs=[gr.outputs.Textbox(label="Generated Text")],
-    title="Text Generation with Grammarly Model"
-)
+additional_inputs=[
+    gr.Slider( label="Temperature", value=0.9, minimum=0.0, maximum=1.0, step=0.05, interactive=True, info="Higher values produce more diverse outputs", ),
+    gr.Slider( label="Max new tokens", value=150, minimum=0, maximum=250, step=64, interactive=True, info="The maximum numbers of new tokens", ),
+    gr.Slider( label="Top-p (nucleus sampling)", value=0.90, minimum=0.0, maximum=1, step=0.05, interactive=True, info="Higher values sample more low-probability tokens", ),
+    gr.Slider( label="Top-k", value=50, minimum=0, maximum=100, step=1, interactive=True, info="Limits the number of top-k tokens considered at each step"),
+]
 
-# Launch the interface
-iface.launch()
+gr.ChatInterface(
+    fn=generate_text,
+    chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
+    additional_inputs=additional_inputs,
+    title="My Grammarly Space",
+    concurrency_limit=20,
+).launch(show_api=False)