update content

app.py

@@ -152,7 +152,7 @@ with gr.Blocks(gr.themes.Soft(), js=js_code_label) as demo:
 
     with gr.Group():
         with gr.Row():
-            with gr.Column(scale=2):
+            with gr.Column(scale=1.5):
                 message = gr.Textbox(label="Prompt", placeholder="Enter your message here")
                 with gr.Row(): 
                     with gr.Column(scale=2):
@@ -166,18 +166,18 @@ with gr.Blocks(gr.themes.Soft(), js=js_code_label) as demo:
                             stop_btn = gr.Button("⏸️ Stop")
                             clear_btn = gr.Button("🔁 Clear")
                         with gr.Row():
-                            gr.Markdown("We thank for the support from [Hyperbolic AI](https://hyperbolic.xyz/).")
+                            gr.Markdown(">> - We thank for the support of Llama-3.1-405B from [Hyperbolic AI](https://hyperbolic.xyz/). ")
             with gr.Column(scale=1):
                 with gr.Accordion("⚙️ Params for **Base** LLM", open=True):
                     with gr.Row():
-                        max_tokens_1 = gr.Slider(label="Max new tokens", value=256, minimum=0, maximum=2048, step=16, interactive=True, visible=True)
+                        max_tokens_1 = gr.Slider(label="Max tokens", value=256, minimum=0, maximum=2048, step=16, interactive=True, visible=True)
                         temperature_1 = gr.Slider(label="Temperature", step=0.01, minimum=0.01, maximum=1.0, value=0.9)
                     with gr.Row():
                         top_p_1 = gr.Slider(label="Top-P", step=0.01, minimum=0.01, maximum=1.0, value=0.9)
                         rp_1 = gr.Slider(label="Repetition Penalty", step=0.1, minimum=0.1, maximum=2.0, value=1.1) 
                 with gr.Accordion("⚙️ Params for **Aligned** LLM", open=True):
                     with gr.Row():
-                        max_tokens_2 = gr.Slider(label="Max new tokens", value=256, minimum=0, maximum=2048, step=16, interactive=True, visible=True)
+                        max_tokens_2 = gr.Slider(label="Max tokens", value=256, minimum=0, maximum=2048, step=16, interactive=True, visible=True)
                         temperature_2 = gr.Slider(label="Temperature", step=0.01, minimum=0.01, maximum=1.0, value=0.9)
                     with gr.Row():
                         top_p_2 = gr.Slider(label="Top-P", step=0.01, minimum=0.01, maximum=1.0, value=0.9)

constant.py

@@ -1,7 +1,9 @@
-HEADER_MD = """# 💬 BaseChat: Chat with Base LLMs by URIAL
+HEADER_MD = """# 💬 BaseChat: Chat with Base LLMs via URIAL
 [📑 Paper](https://arxiv.org/abs/2312.01552) | [🛜 Website](https://allenai.github.io/re-align/) | [💻 GitHub](https://github.com/Re-Align/urial) | [🐦 X-1](https://x.com/billyuchenlin/status/1799885923045568531) [🐦 X-2](https://x.com/billyuchenlin/status/1759541978881311125) | 📮 Contact: [Yuchen Lin](https://yuchenlin.xyz/)
 
 **Talk with __BASE__ LLMs which are not fine-tuned at all. The used URIAL prompt is [here](https://github.com/Re-Align/URIAL/blob/main/urial_prompts/inst_1k_v4.help.txt.md).** 
+- We now also show the responses of the associated aligned model for comparisons.
+
 """
 
 js_code_label = """