Update app.py

app.py

@@ -83,6 +83,7 @@ pwd_aoai_url = PasswordInput(
 def check_yi_mode(event):  
     # Make openai_row invisible when local YI Mode is enabled and vice versa.  
     openai_row.visible = not event.new  
+    YI_ENDPOINT.visible = event.new  
 
 YI_MODE = pn.widgets.Checkbox(name='Local Y.I. Mode', value=True)     
 YI_ENDPOINT = pn.widgets.TextInput(name='Yi Endpoint', placeholder="Enter your YI endpoint here...")