Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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"),
|