tatianija commited on
Commit
9c5a793
·
verified ·
1 Parent(s): 045c4b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -475,6 +475,7 @@ def start_answer_generation(model_choice: str):
475
  # Map model choice to actual model name
476
  model_map = {
477
  "Llama 3.1 8B": "meta-llama/Llama-3.1-8B-Instruct",
 
478
  "Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3"
479
  }
480
 
@@ -682,7 +683,7 @@ with gr.Blocks(title="Intelligent Agent with Media Processing") as demo:
682
 
683
  with gr.Row():
684
  model_choice = gr.Dropdown(
685
- choices=["Llama 3.1 8B", "Mistral 7B"],
686
  value="Llama 3.1 8B",
687
  label="Select Model"
688
  )
 
475
  # Map model choice to actual model name
476
  model_map = {
477
  "Llama 3.1 8B": "meta-llama/Llama-3.1-8B-Instruct",
478
+ "Llama 3.3 70B": "meta-llama/Llama-3.3-70B-Instruct",
479
  "Mistral 7B": "mistralai/Mistral-7B-Instruct-v0.3"
480
  }
481
 
 
683
 
684
  with gr.Row():
685
  model_choice = gr.Dropdown(
686
+ choices=["Llama 3.1 8B", "Llama 3.3 70B", "Mistral 7B"],
687
  value="Llama 3.1 8B",
688
  label="Select Model"
689
  )