Update app.py

app.py

@@ -90,12 +90,14 @@ YI_ENDPOINT = pn.widgets.TextInput(name='Yi Endpoint', placeholder="Enter your Y
 # Add an observer to watch changes in yi_mode value (True/False)    
 YI_MODE.param.watch(check_yi_mode, 'value')    
 
-pn.Column(  
+
+yi_widgets = pn.Column(  
   pn.pane.Markdown("### Local YI Mode: "),   
   YI_MODE,   
   pn.pane.Markdown("### Yi Endpoint: "),   
   YI_ENDPOINT  
 )  
+template.main.append(yi_widgets)
 openai_row = pn.Row(txt_model, pwd_openai_key, pwd_openai_url)  
 file_cfg = pn.widgets.FileInput(filename="OAI_CONFIG_LIST", sizing_mode="stretch_width")
 template.main.append(