Update with h2oGPT hash 5bf309e905fe1b771229dd0f61b0582ff883b41d

app.py

@@ -648,6 +648,14 @@ body.dark{background:linear-gradient(#0d0d0d,#333333);}"""
 
         Chatbot._postprocess_chat_messages = _postprocess_chat_messages
 
+    dark_js = """() => {
+        if (document.querySelectorAll('.dark').length) {
+            document.querySelectorAll('.dark').forEach(el => el.classList.remove('dark'));
+        } else {
+            document.querySelector('body').classList.add('dark');
+        }
+    }"""
+
     demo = gr.Blocks(theme=gr.themes.Soft(**colors_dict), css=css_code, title="h2oGPT", analytics_enabled=False)
     callback = gr.CSVLogger()
     # css_code = 'body{background-image:url("https://h2o.ai/content/experience-fragments/h2o/us/en/site/header/master/_jcr_content/root/container/header_copy/logo.coreimg.svg/1678976605175/h2o-logo.svg");}'
@@ -737,7 +745,7 @@ body.dark{background:linear-gradient(#0d0d0d,#333333);}"""
                                     lines=4, label=instruction_label,
                                     placeholder=kwargs['placeholder_instruction'],
                                 )
-                            with gr.Row():  # .style(equal_height=False, equal_width=False):
+                            with gr.Row():
                                 submit = gr.Button(value='Submit').style(full_width=False, size='sm')
                                 stop_btn = gr.Button(value="Stop").style(full_width=False, size='sm')
                         with gr.Row():
@@ -841,7 +849,7 @@ body.dark{background:linear-gradient(#0d0d0d,#333333);}"""
                         n_gpus = torch.cuda.device_count()
                         n_gpus_list = [str(x) for x in list(range(-1, n_gpus))]
                         with gr.Column():
-                            with gr.Row(scale=1):
+                            with gr.Row():
                                 with gr.Column(scale=50):
                                     model_choice = gr.Dropdown(model_options_state.value[0], label="Choose Model",
                                                                value=kwargs['base_model'])
@@ -860,7 +868,7 @@ body.dark{background:linear-gradient(#0d0d0d,#333333);}"""
                                     model_used = gr.Textbox(label="Current Model", value=kwargs['base_model'])
                                     lora_used = gr.Textbox(label="Current LORA", value=kwargs['lora_weights'],
                                                            visible=kwargs['show_lora'])
-                            with gr.Row(scale=1):
+                            with gr.Row():
                                 with gr.Column(scale=50):
                                     new_model = gr.Textbox(label="New Model HF name/path")
                                     new_lora = gr.Textbox(label="New LORA HF name/path", visible=kwargs['show_lora'])
@@ -869,7 +877,7 @@ body.dark{background:linear-gradient(#0d0d0d,#333333);}"""
                                     add_lora_button = gr.Button("Add new LORA name", visible=kwargs['show_lora'])
                         col_model2 = gr.Column(visible=False)
                         with col_model2:
-                            with gr.Row(scale=1):
+                            with gr.Row():
                                 with gr.Column(scale=50):
                                     model_choice2 = gr.Dropdown(model_options_state.value[0], label="Choose Model 2",
                                                                 value=no_model_str)
@@ -944,13 +952,7 @@ body.dark{background:linear-gradient(#0d0d0d,#333333);}"""
             None,
             None,
             None,
-            _js="""() => {
-            if (document.querySelectorAll('.dark').length) {
-                document.querySelectorAll('.dark').forEach(el => el.classList.remove('dark'));
-            } else {
-                document.querySelector('body').classList.add('dark');
-            }
-        }""",
+            _js=dark_js,
             api_name="dark",
         )
 
@@ -1414,6 +1416,7 @@ body.dark{background:linear-gradient(#0d0d0d,#333333);}"""
         stop_btn.click(lambda: None, None, None,
                        cancels=[submit_event_nochat, submit_event, submit_event2, submit_event3],
                        queue=False, api_name='stop').then(clear_torch_cache)
+        demo.load(None,None,None,_js=dark_js)
 
     demo.queue(concurrency_count=1)
     favicon_path = "h2o-logo.svg"