tatianija commited on
Commit
f0b3ee7
·
verified ·
1 Parent(s): d57cf0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -958,7 +958,7 @@ def start_answer_generation(model_choice: str):
958
  model_map = {
959
  "Llama 3.1 8B": "meta-llama/Llama-3.1-8B-Instruct",
960
  "Llama 3.3 70B": "meta-llama/Llama-3.3-70B-Instruct",
961
- "Llama shallow": "tokyotech-llm/Llama-3.3-Swallow-70B-Instruct-v0.4",
962
  "Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3",
963
  "Qwen 2.5": "Qwen/Qwen‑2.5‑Omni‑7B",
964
  #"Qwen 2.5 instruct": "Qwen/Qwen2.5-14B-Instruct-1M",
@@ -1117,7 +1117,7 @@ with gr.Blocks(title="Intelligent Agent with Media Processing") as demo:
1117
 
1118
  with gr.Row():
1119
  model_choice = gr.Dropdown(
1120
- choices=["Llama 3.1 8B", "Llama 3.3 70B", "Llama shallow", "Mistral 7B", "Qwen 2.5", "Qwen 3"],
1121
  value="Llama 3.1 8B",
1122
  label="Select Model"
1123
  )
 
958
  model_map = {
959
  "Llama 3.1 8B": "meta-llama/Llama-3.1-8B-Instruct",
960
  "Llama 3.3 70B": "meta-llama/Llama-3.3-70B-Instruct",
961
+ "Llama 3.3 Shallow 70B": "tokyotech-llm/Llama-3.3-Swallow-70B-Instruct-v0.4",
962
  "Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3",
963
  "Qwen 2.5": "Qwen/Qwen‑2.5‑Omni‑7B",
964
  #"Qwen 2.5 instruct": "Qwen/Qwen2.5-14B-Instruct-1M",
 
1117
 
1118
  with gr.Row():
1119
  model_choice = gr.Dropdown(
1120
+ choices=["Llama 3.1 8B", "Llama 3.3 70B", "Llama 3.3 Shallow 70B", "Mistral 7B", "Qwen 2.5", "Qwen 3"],
1121
  value="Llama 3.1 8B",
1122
  label="Select Model"
1123
  )