Update app.py

app.py

@@ -53,12 +53,17 @@ def generate_text(prompt):
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("<h1>🧠 Phi-2 QLoRA Text Generator</h1>")
     
-    # Textbox for user input and a button to generate text
+    # Textbox for user input
     prompt = gr.Textbox(label="Enter your prompt:", lines=2)
+    
+    # Output textbox for generated text
     output = gr.Textbox(label="Generated text:", lines=5)
+    
+    # Button to trigger text generation
+    generate_button = gr.Button("Generate Text")
 
-    # Generate text when the button is clicked
-    prompt.submit(generate_text, prompt, output)
+    # Set the button action to generate text
+    generate_button.click(generate_text, inputs=prompt, outputs=output)
 
 # Launch the app
 demo.launch(share=True)