0.42 Layout tweaks.

app.py

@@ -42,7 +42,7 @@ try:
         device_map="auto",
         trust_remote_code=True,
     )
-    #model_a.tie_weights()
+
     tokenizer_b = AutoTokenizer.from_pretrained(model_info[1]['id'])
     model_b = AutoModelForCausalLM.from_pretrained(
         model_info[1]['id'],
@@ -50,7 +50,7 @@ try:
         device_map="auto",
         trust_remote_code=True,
     )
-    model_b.tie_weights()
+    
 except Exception as e:
     logging.error(f'{SPACER} Error: {e}, Traceback {traceback.format_exc()}')
 
@@ -197,17 +197,12 @@ def reveal_bot(selection, chatbot_a, chatbot_b):
         chatbot_b.append(["🤝", f"Dunno. I am {model_info[1]['name']}"])            
     return chatbot_a, chatbot_b
 
-arena_notes = """## Important Notes:
-- Sometimes an error may occur when generating the response, in this case, please try again.
-"""
-
 with gr.Blocks() as demo:
     try:
         with gr.Column():
-            gr.HTML("<center><h1>🤖le Royale</h1></center>")
-            gr.Markdown(arena_notes)
-            with gr.Row(variant="panel"):
-                with gr.Column(scale=1):
+            gr.HTML("<center><h1>Bot Battle Royale</h1><p>Try a prompt in a language you like. Set the parameters and vote for the best answers. Finally, the bots reveal their identity.</p></center>")
+            with gr.Row(variant="compact"):
+                with gr.Column(scale=0):
                     language_dropdown = gr.Dropdown(
                         choices=["English", "German", "French", "Spanish"], 
                         label="Select Language for System Prompt",
@@ -233,9 +228,9 @@ with gr.Blocks() as demo:
             with gr.Row(variant="panel"):
                     better_bot = gr.Radio(["Bot A kicks ass!", "Bot B crushes it!", "It's a draw."], label="Rate the output!")
             with gr.Accordion(label="Generation Configurations", open=False):
-                max_new_tokens = gr.Slider(minimum=128, maximum=4096, value=2048, label="Max New Tokens", step=128)
+                max_new_tokens = gr.Slider(minimum=128, maximum=4096, value=512, label="Max new tokens", step=128)
                 temperature = gr.Slider(minimum=0.0, maximum=1.0, value=0.7, label="Temperature", step=0.01)
-                top_p = gr.Slider(minimum=0.0, maximum=1.0, value=1.0, label="Top-p", step=0.01)
+                top_p = gr.Slider(minimum=0.0, maximum=1.0, value=1.0, label="Top_p", step=0.01)
                 repetition_penalty = gr.Slider(minimum=0.1, maximum=2.0, value=1.1, label="Repetition Penalty", step=0.1)
 
 
@@ -245,7 +240,7 @@ with gr.Blocks() as demo:
             outputs=[system_prompt]
         )
 
-        better_bot.select(reveal_bot, inputs=[better_bot, chatbot_a, chatbot_b], outputs=[chatbot_a, chatbot_b]) #fckp outputs=[chatbot_a, chatbot_b]
+        better_bot.select(reveal_bot, inputs=[better_bot, chatbot_a, chatbot_b], outputs=[chatbot_a, chatbot_b]) 
         input_text.submit(generate_both, inputs=[system_prompt, input_text, chatbot_a, chatbot_b, max_new_tokens, temperature, top_p, repetition_penalty], outputs=[chatbot_a, chatbot_b])
         submit_btn.click(generate_both, inputs=[system_prompt, input_text, chatbot_a, chatbot_b, max_new_tokens, temperature, top_p, repetition_penalty], outputs=[chatbot_a, chatbot_b])
         clear_btn.click(clear, outputs=[chatbot_a, chatbot_b])