jonaschua commited on
Commit
93116a8
·
verified ·
1 Parent(s): 10ae4aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -20,7 +20,7 @@ For more information on `huggingface_hub` Inference API support, please check th
20
 
21
 
22
  def choose_model(model_name):
23
- if model_name == "Qwen1.5":
24
  model = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
25
 
26
  elif model_name == "Llama3-8b-Instruct":
@@ -37,7 +37,9 @@ def choose_model(model_name):
37
 
38
  elif model_name == "Mixtral-8x7B-Instruct":
39
  model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
40
-
 
 
41
 
42
  return model
43
 
@@ -79,7 +81,7 @@ demo = gr.ChatInterface(
79
  respond,
80
 
81
  additional_inputs=[
82
- gr.Dropdown(["Qwen1.5", "Gemma-2-2b", "Llama2-13b-chat", "Llama3-8b-Instruct", "Llama3.1-8b-Instruct", "Mixtral-8x7B-Instruct"], label="Select Model"),
83
  gr.Textbox(value="You are a friendly and helpful Chatbot, be concise and straight to the point, avoid excessive reasoning.", label="System message"),
84
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
85
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
 
20
 
21
 
22
  def choose_model(model_name):
23
+ if model_name == "DeepSeek-R1-Distill-Qwen-1.5B":
24
  model = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
25
 
26
  elif model_name == "Llama3-8b-Instruct":
 
37
 
38
  elif model_name == "Mixtral-8x7B-Instruct":
39
  model = "mistralai/Mixtral-8x7B-Instruct-v0.1"
40
+
41
+ elif model_name == "Zephr-7b-beta":
42
+ model = "HuggingFaceH4/zephyr-7b-beta"
43
 
44
  return model
45
 
 
81
  respond,
82
 
83
  additional_inputs=[
84
+ gr.Dropdown(["DeepSeek-R1-Distill-Qwen-1.5B", "Gemma-2-2b", "Llama2-13b-chat", "Llama3-8b-Instruct", "Llama3.1-8b-Instruct", "Mixtral-8x7B-Instruct", "Zephr-7b-beta"], label="Select Model"),
85
  gr.Textbox(value="You are a friendly and helpful Chatbot, be concise and straight to the point, avoid excessive reasoning.", label="System message"),
86
  gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
87
  gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),