Update app.py

app.py

@@ -106,7 +106,7 @@ with gr.Blocks() as app:
                     with gr.Column(scale=1):
                         stop_btn = gr.Button("Stop")
                         clear_btn = gr.Button("Clear")
-                client_choice = gr.Dropdown(label="Models", type='index', choices=[c for c in models if c in [client.config.model_id for client in clients]], value=models[0], interactive=True)
+                client_choice = gr.Dropdown(label="Models", type='index', choices=models, value=models[0], interactive=True)
             with gr.Column(scale=1):
                 rand = gr.Checkbox(label="Random Seed", value=True)
                 seed = gr.Slider(label="Seed", minimum=1, maximum=1111111111111111, step=1, value=rand_val)
@@ -122,7 +122,7 @@ with gr.Blocks() as app:
             im_width = gr.Number(label="Width", value=500)
             wait_time = gr.Number(label="Wait Time", value=3000)
             theme = gr.Radio(label="Theme", choices=["light", "dark"], value="light")
-            chatblock = gr.Dropdown(label="Chatblocks", choices=[c for c in range(0, 21)], value=0, type="index")
+            chatblock = gr.Dropdown(label="Chatblocks", choices=list(range(0, 21)), value=0, type="index")
             header = gr.Checkbox(label="Include header?", value=True)
     rand.change(check_rand, inputs=[rand, seed], outputs=seed)
     btn.click(chat_inf, inputs=[sys_inp, inp, chat_b, memory, client_choice, seed, temp, tokens, top_p, rep_p, chat_mem, custom_prompt], outputs=[chat_b, memory])