Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 == "
|
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(["
|
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"),
|