0.35 chatch exception

app.py

@@ -190,32 +190,35 @@ arena_notes = """## Important Notes:
 """
 
 with gr.Blocks() as demo:
-    with gr.Column():
-        gr.HTML("<center><h1>🤖le Royale</h1></center>")
-        gr.Markdown(arena_notes)
-        system_prompt = gr.Textbox(lines=1, label="System Prompt", value="You are a helpful chatbot that adheres to the prompted request.", show_copy_button=True)
-        with gr.Row(variant="panel"):
-            with gr.Column(scale=1):
-                submit_btn = gr.Button(value="Generate", variant="primary")
-                clear_btn = gr.Button(value="Clear", variant="secondary")
-            input_text = gr.Textbox(lines=1, label="Prompt", value="Write a Nike style ad headline about the shame of being second best.", scale=3, show_copy_button=True)
-        with gr.Row(variant="panel"):
-            with gr.Column():
-                chatbot_a = gr.Chatbot(label="Model A", show_copy_button=True, height=500)
-            with gr.Column():
-                chatbot_b = gr.Chatbot(label="Model B", show_copy_button=True, height=500)
-        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)
-            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)
-            repetition_penalty = gr.Slider(minimum=0.1, maximum=2.0, value=1.1, label="Repetition Penalty", step=0.1)
-
-    better_bot.select(reveal_bot, inputs=[better_bot, chatbot_a, chatbot_b], outputs=[chatbot_a, chatbot_b]) #fckp 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])
+    try:
+        with gr.Column():
+            gr.HTML("<center><h1>🤖le Royale</h1></center>")
+            gr.Markdown(arena_notes)
+            system_prompt = gr.Textbox(lines=1, label="System Prompt", value="You are a helpful chatbot that adheres to the prompted request.", show_copy_button=True)
+            with gr.Row(variant="panel"):
+                with gr.Column(scale=1):
+                    submit_btn = gr.Button(value="Generate", variant="primary")
+                    clear_btn = gr.Button(value="Clear", variant="secondary")
+                input_text = gr.Textbox(lines=1, label="Prompt", value="Write a Nike style ad headline about the shame of being second best.", scale=3, show_copy_button=True)
+            with gr.Row(variant="panel"):
+                with gr.Column():
+                    chatbot_a = gr.Chatbot(label="Model A", show_copy_button=True, height=500)
+                with gr.Column():
+                    chatbot_b = gr.Chatbot(label="Model B", show_copy_button=True, height=500)
+            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)
+                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)
+                repetition_penalty = gr.Slider(minimum=0.1, maximum=2.0, value=1.1, label="Repetition Penalty", step=0.1)
+
+        better_bot.select(reveal_bot, inputs=[better_bot, chatbot_a, chatbot_b], outputs=[chatbot_a, chatbot_b]) #fckp 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])
+    except Exception as e:
+        logging.error(f'{SPACER} Error: {e}, Traceback {traceback.format_exc()}')
 
 if __name__ == "__main__":
     demo.queue().launch()