Update app.py

app.py

@@ -84,8 +84,8 @@ file_cfg = pn.widgets.FileInput(filename="OAI_CONFIG_LIST", sizing_mode="stretch
 template.main.append(  
   pn.Column(  
       openai_row,  
-      [pn.pane.Markdown("### Local YI Mode: "),Yi_Mode],
-      [pn.pane.Markdown("### Yi Endpoint: "), yi_endpoint]  
+      [pn.pane.Markdown("### Local YI Mode: "),YI_MODE],
+      [pn.pane.Markdown("### Yi Endpoint: "), YI_ENDPOINT]  
     )
 )
 
@@ -97,7 +97,6 @@ def check_yi_mode(event):
     # Make openai_row invisible when local YI Mode is enabled and vice versa.  
     openai_row.visible = not event.new  
   
-# Add an observer to watch changes in yi_mode value (True/False)  
 YI_MODE.param.watch(check_yi_mode, 'value')  
 
 def get_config(tmpfilename="OAI_CONFIG_LIST"):
@@ -118,7 +117,7 @@ def get_config(tmpfilename="OAI_CONFIG_LIST"):
         {
             "model": "EasyYI",  
             "api_key": "None",
-            "base_url": "https://huggingface.co/spaces/teamtonic/easyyi"
+            "base_url": YI_ENDPOINT
         }
     ] 
     if not config_list: