Update app.py

app.py

@@ -255,6 +255,7 @@ description = """
      <img src="https://huggingface.co/datasets/trl-internal-testing/example-images/resolve/main/images/textenv_demo_banner.png">
 </div>
 <div style="text-align: left;">
+    <hr>
     <p>This is a demo to generate text and code with the following StarCoderBase models:</p>
     <ul>
         <li><a href="https://huggingface.co/vwxyzjn/starcoderbase-triviaqa">StarCoderBase TriviaQA</a>: A finetuned version of StarCoderBase on on the TriviaQA dataset using reinforcement learning via <a href="https://github.com/huggingface/trl/pull/424">TRL's TextEnvironment</a>.</li>
@@ -284,14 +285,16 @@ with gr.Blocks(theme=theme, analytics_enabled=False, css=css) as demo:
                     elem_id="q-input",
                 )
                 submit = gr.Button("Generate", variant="primary")
-                #
+                
                 output = gr.HighlightedText(
                     label="Output",
                     color_map={"query": "red", "call": "green", "response": "blue", "submit": "yellow", "model": "pink"},
                 )
-                output2 = gr.Code(elem_id="q-output", lines=30, label="Raw output")
+               
                 with gr.Row():
                     with gr.Column():
+                        with gr.Accordion("Raw output", open=False):
+                            output2 = gr.Code(elem_id="q-output", lines=30, label="Raw output")
                         with gr.Accordion("Advanced settings", open=False):
                             with gr.Row():
                                 column_1, column_2 = gr.Column(), gr.Column()