Update app.py

app.py

@@ -1,8 +1,24 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-def chat_inf(system_prompt,prompt,history):
+models=[
+    "google/gemma-7b",
+    "google/gemma-7b-it",
+    "google/gemma-2b",
+    "google/gemma-2b-it"
+]
+clients=[
+InferenceClient(models[0]),
+InferenceClient(models[1]),
+InferenceClient(models[2]),
+InferenceClient(models[3]),
+]
+
+
+
+
+def chat_inf(system_prompt,prompt,history,client_choice):
+    client=clients[int(client_choice)-1]
     if not history:
         history = []
         hist_len=0
@@ -42,6 +58,6 @@ with gr.Blocks() as app:
                 with gr.Group():
                     stop_btn=gr.Button("Stop")
                     clear_btn=gr.Button("Clear")
-        chatblock=gr.Dropdown(label="Chatblocks",choices=[c for c in range(1,40)],multiselect=True)
+        chatblock=gr.Dropdown(label="Chatblocks",type='index',choices=[c for c in models],value=1)
     btn.click(chat_inf,[sys_inp,inp,chat_b],chat_b)
 app.launch()