Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -323,9 +323,7 @@ def start_answer_generation(model_choice: str):
|
|
| 323 |
# Map model choice to actual model name
|
| 324 |
model_map = {
|
| 325 |
"Llama 3.1 8B": "meta-llama/Llama-3.1-8B-Instruct",
|
| 326 |
-
"
|
| 327 |
-
"Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3",
|
| 328 |
-
"CodeLlama 7B": "codellama/CodeLlama-7b-Instruct-hf"
|
| 329 |
}
|
| 330 |
|
| 331 |
selected_model = model_map.get(model_choice, "meta-llama/Llama-3.1-8B-Instruct")
|
|
|
|
| 323 |
# Map model choice to actual model name
|
| 324 |
model_map = {
|
| 325 |
"Llama 3.1 8B": "meta-llama/Llama-3.1-8B-Instruct",
|
| 326 |
+
"Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3"
|
|
|
|
|
|
|
| 327 |
}
|
| 328 |
|
| 329 |
selected_model = model_map.get(model_choice, "meta-llama/Llama-3.1-8B-Instruct")
|