jonaschua commited on
Commit
1aea96c
·
verified ·
1 Parent(s): 99dd0b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -38,6 +38,9 @@ def choose_model(model_name):
38
  elif model_name == "Gemma-2-2b":
39
  model = "google/gemma-2-2b-it"
40
 
 
 
 
41
  elif model_name == "Mixtral-8x7B-Instruct":
42
  model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
43
 
@@ -87,7 +90,7 @@ demo = gr.ChatInterface(
87
  respond,
88
 
89
  additional_inputs=[
90
- gr.Dropdown(["DeepSeek-R1-Distill-Qwen-1.5B", "DeepSeek-R1-Distill-Qwen-32B", "Gemma-2-2b", "Llama2-13b-chat", "Llama3-8b-Instruct", "Llama3.1-8b-Instruct", "Microsoft-phi-2", "Mixtral-8x7B-Instruct", "Zephr-7b-beta"], label="Select Model"),
91
  gr.Textbox(value="You are a friendly and helpful Chatbot, be concise and straight to the point, avoid excessive reasoning.", label="System message"),
92
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
93
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
 
38
  elif model_name == "Gemma-2-2b":
39
  model = "google/gemma-2-2b-it"
40
 
41
+ elif model_name == "Gemma-7b":
42
+ model = "google/gemma-7b"
43
+
44
  elif model_name == "Mixtral-8x7B-Instruct":
45
  model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
46
 
 
90
  respond,
91
 
92
  additional_inputs=[
93
+ gr.Dropdown(["DeepSeek-R1-Distill-Qwen-1.5B", "DeepSeek-R1-Distill-Qwen-32B", "Gemma-2-2b", "Gemma-7b", "Llama2-13b-chat", "Llama3-8b-Instruct", "Llama3.1-8b-Instruct", "Microsoft-phi-2", "Mixtral-8x7B-Instruct", "Zephr-7b-beta"], label="Select Model"),
94
  gr.Textbox(value="You are a friendly and helpful Chatbot, be concise and straight to the point, avoid excessive reasoning.", label="System message"),
95
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
96
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),