Update app.py

app.py

@@ -51,19 +51,18 @@ with gr.Blocks() as demo:
     <h1 style="text-align: center;">Legal Document Drafting</h1>
     """)
     with gr.Tab("Writing Assist"):
-#        with gr.Column():
-            result = gr.components.Textbox(lines=7, label="Writing Assist", placeholder=prompts[0])
-            prompt = gr.components.Textbox(lines=2, label="Prompt", placeholder=examples[0], visible=False)
-            gr.Examples(examples, label='Examples', inputs=[prompt])
-            prompt.change(generate, inputs=[prompt], outputs=[result])
-            btn = gr.Button("Next sentence")
-            btn.click(generate, inputs=[result], outputs=[result])
-#        with gr.Column():
+        result = gr.components.Textbox(lines=7, label="Writing Assist", placeholder=prompts[0])
+        prompt = gr.components.Textbox(lines=2, label="Prompt", placeholder=examples[0], visible=False)
+        gr.Examples(examples, label='Examples', inputs=[prompt])
+        prompt.change(generate, inputs=[prompt], outputs=[result])
+        btn = gr.Button("Next sentence")
+        btn.click(generate, inputs=[result], outputs=[result])
     with gr.Tab("Random Generative"):
-            result2 = gr.components.Textbox(lines=7, label="Random Generative", show_label=True, placeholder=prompts[1])
-            gr.Examples(examples, label='Examples', inputs=[result2])
-            rnd_btn = gr.Button("Random Drafting")
-            rnd_btn.click(rnd_generate, inputs=[result2], outputs=[result2])
+        result2 = gr.components.Textbox(lines=7, label="Random Generative", show_label=True, placeholder=prompts[1])
+        gr.Examples(examples, label='Examples', inputs=[result2])
+        rnd_btn = gr.Button("Random Drafting")
+        rnd_btn.click(rnd_generate, inputs=[result2], outputs=[result2])
+    gr.Markdown("Legal Document drafting demo")
     
 if __name__ == "__main__":
     demo.launch()