Update app.py

app.py

@@ -22,15 +22,15 @@ def generate_text(prompt, max_new_tokens, top_k, top_p, do_sample):
 def main():
     with gr.Blocks() as demo:
         gr.Markdown(title)
-        output = gr.Textbox(label="Generated Text", lines=10)
-        submit_button = gr.Button("Generate")
-        with gr.Accordion("Optional Parameters"):
-            max_new_tokens = gr.Slider(minimum=1, maximum=650, step=1, value=300, label="new tokens in reply")
-            top_k = gr.Slider(minimum=1, maximum=100, step=1, value=50, label="Top K")
-            top_p = gr.Slider(minimum=0.1, maximum=1.0, step=0.01, value=0.95, label="Top P")
-            do_sample = gr.Checkbox(value=True, label="Untick for faster inference")
+        output = gr.Textbox(label="👴🏻🏑OLMo", lines=10)
+        submit_button = gr.Button("🌟Try it Out!")
+        with gr.Accordion("🤖Optional Parameters"):
+            max_new_tokens = gr.Slider(minimum=1, maximum=650, step=1, value=300, label="🪙new tokens in reply")
+            top_k = gr.Slider(minimum=1, maximum=100, step=1, value=50, label="🗣️Top K")
+            top_p = gr.Slider(minimum=0.1, maximum=1.0, step=0.01, value=0.95, label="🤫Repetition Penalty")
+            do_sample = gr.Checkbox(value=False, label="Untick for faster inference")
         prompt = gr.Textbox(label="Enter your prompt")
-        prompt.change(fn=generate_text, inputs=[prompt, max_new_tokens, top_k, top_p, do_sample], outputs=output, submit=submit_button)
+        submit_button.click(fn=generate_text, inputs=[prompt, max_new_tokens, top_k, top_p, do_sample], outputs=output)
 
     demo.launch()