Update app.py

app.py

@@ -27,9 +27,15 @@ def choose_model(model_name):
     elif model_name == "Llama3.1-8b-Instruct":
         model = "meta-llama/Llama-3.1-8B-Instruct"
 
-    elif model_name == "Llama2-13b-chat"
+    elif model_name == "Llama2-13b-chat":
         model = "meta-llama/Llama-2-13b-chat-hf"
 
+    elif model_name == "Gemma-2-2b":
+        model = "google/gemma-2-2b-it"
+
+    elif model_name == "Mixtral-8x7B-Instruct":
+        model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
+    
     client = InferenceClient(model, token=os.getenv('deepseekv2'))
     return client
     
@@ -68,7 +74,7 @@ demo = gr.ChatInterface(
     respond,
 
     additional_inputs=[
-        gr.Dropdown(["Qwen1.5", "Llama2-13b-chat", "Llama3-8b-Instruct", "Llama3.1-8b-Instruct"], label="Select Model"),
+        gr.Dropdown(["Qwen1.5", "Gemma-2-2b", "Llama2-13b-chat", "Llama3-8b-Instruct", "Llama3.1-8b-Instruct", "Mixtral-8x7B-Instruct"], label="Select Model"),
         gr.Textbox(value="You are a friendly and helpful Chatbot, be concise and straight to the point, avoid excessive reasoning.", label="System message"),
         gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),