:zap: add clear fn

app.py

@@ -5,7 +5,7 @@ import gradio as gr
 import torch
 from transformers import pipeline
 
-from utils import postprocess
+from utils import postprocess, clear
 
 logging.basicConfig(
     level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
@@ -136,13 +136,18 @@ if __name__ == "__main__":
                 label="Email Prompt",
                 value=default_prompt,
             )
-            num_gen_tokens = gr.Slider(
-                label="Generation Tokens",
-                value=64,
-                maximum=128,
-                minimum=32,
-                step=16,
-            )
+
+            with gr.Row():
+                clear_button = gr.Button(
+                    value="Clear Prompt",
+                )
+                num_gen_tokens = gr.Slider(
+                    label="Generation Tokens",
+                    value=64,
+                    maximum=128,
+                    minimum=32,
+                    step=16,
+                )
 
             generated_email = gr.Textbox(
                 label="Generated Result",
@@ -155,7 +160,7 @@ if __name__ == "__main__":
 
             gr.Markdown("## Advanced Options")
             gr.Markdown(
-                "See details about these parameters [here](https://huggingface.co/blog/how-to-generate), otherwise they should be fine as-is."
+                "This demo generates text via beam search. See details about these parameters [here](https://huggingface.co/blog/how-to-generate), otherwise they should be fine as-is."
             )
 
             num_beams = gr.Radio(
@@ -183,7 +188,11 @@ if __name__ == "__main__":
                 "The intended use of this model is to provide suggestions to _auto-complete_ the rest of your email. Said another way, it should serve as a **tool to write predictable emails faster**. It is not intended to write entire emails from scratch; at least **some input** is required to guide the direction of the model.\n\nPlease verify any suggestions by the model for A) False claims and B) negation statements **before** accepting/sending something."
             )
             gr.Markdown("---")
-
+        clear_button.on_click(
+            fn=clear,
+            input_text=[prompt_text],
+            output_text=[prompt_text],
+        )
         generate_button.click(
             fn=generate_text,
             inputs=[