Update app.py

app.py

@@ -41,8 +41,8 @@ def respond(prompt, max_length, min_length, max_new_tokens, num_beams, temperatu
     #response = correct_text(prompt, max_length, max_new_tokens, min_length, num_beams, temperature, top_p)
     return prompt
 
-def update_prompt(text):
-    return text
+def update_prompt(promptBtn):
+    return promptBtn.value
 
 # Create the Gradio interface
 with gr.Blocks() as demo:
@@ -56,13 +56,13 @@ with gr.Blocks() as demo:
 
     # Sample prompts
     with gr.Row():
-        samp1 = gr.Button(value="we shood buy an car")
-        samp2 = gr.Button(value="she is more taller")
-        samp3 = gr.Button(value="John and i saw a sheep over their.")
+        samp1 = gr.Button("we shood buy an car")
+        samp2 = gr.Button("she is more taller")
+        samp3 = gr.Button("John and i saw a sheep over their.")
         
-        samp1.click(update_prompt, samp1.value, prompt_box)
-        samp2.click(update_prompt, samp2.value, prompt_box)
-        samp3.click(update_prompt, samp3.value, prompt_box)
+        samp1.click(update_prompt, samp1, prompt_box)
+        samp2.click(update_prompt, samp2, prompt_box)
+        samp3.click(update_prompt, samp3, prompt_box)
         
     
     with gr.Accordion("Generation Parameters:", open=False):