tatianija commited on
Commit
291590a
·
verified ·
1 Parent(s): f91b897

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
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
- "Llama 3.1 70B": "meta-llama/Llama-3.1-70B-Instruct",
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")