Update app.py

app.py

@@ -221,7 +221,7 @@ def chat_inf_d(system_prompt,prompt,history,client_choice,seed,temp,tokens,top_p
         yield None        
 def add_new_model(inp, cur):
     cur.append(inp)
-    return cur
+    return cur,gr.update(choices=[z for z in cur])
 def load_new(models=models):
     return models
     
@@ -255,7 +255,7 @@ with gr.Blocks() as app:
                         with gr.Group():
                             stop_btn=gr.Button("Stop")
                             clear_btn=gr.Button("Clear")                
-                client_choice=gr.Dropdown(label="Models",type='index',choices=[c for c in new_models],max_choices=4,multiselect=True,interactive=True)
+                client_choice=gr.Dropdown(label="Models",type='index',choices=[c for c in models],max_choices=4,multiselect=True,interactive=True)
                 add_model=gr.Textbox(label="New Model")
                 add_btn=gr.Button("Add Model")
             with gr.Column(scale=1):
@@ -284,7 +284,7 @@ with gr.Blocks() as app:
     hid4=gr.Number(value=4,visible=False)    
 
     app.load(load_new,None,new_models)
-    add_btn.click(add_new_model,[add_model,new_models],new_models)
+    add_btn.click(add_new_model,[add_model,new_models],[new_models,client_choice])
     client_choice.change(load_models,client_choice,[chat_a,chat_b,chat_c,chat_d])
 
     #im_go=im_btn.click(get_screenshot,[chat_b,im_height,im_width,chatblock,theme,wait_time],img)