pmlocek commited on
Commit
8983b86
1 Parent(s): faef657

Update models.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -293,7 +293,7 @@ def main():
293
 
294
  navigator_llm = st.selectbox(
295
  "Navigator LLM",
296
- options=["gretelai/gpt-auto", "gretelai/gpt-llama3-8b"],
297
  index=0,
298
  help="Select the primary Navigator LLM",
299
  )
@@ -319,7 +319,7 @@ def main():
319
  "Select additional Navigator LLMs for co-teaching in AAA. It is recommended to use different LLMs than the primary Navigator LLM for this step."
320
  )
321
 
322
- co_teach_options = ["gretelai/gpt-llama3-8b", "gretelai/gpt-mistral7b"]
323
  for model in co_teach_options:
324
  if st.checkbox(model, value=True, key=f"checkbox_{model}"):
325
  co_teach_llms.append(model)
 
293
 
294
  navigator_llm = st.selectbox(
295
  "Navigator LLM",
296
+ options=["gretelai/gpt-auto", "gretelai/gpt-llama3-1-8b"],
297
  index=0,
298
  help="Select the primary Navigator LLM",
299
  )
 
319
  "Select additional Navigator LLMs for co-teaching in AAA. It is recommended to use different LLMs than the primary Navigator LLM for this step."
320
  )
321
 
322
+ co_teach_options = ["gretelai/gpt-llama3-1-8b", "gretelai/gpt-mistral-nemo-2407"]
323
  for model in co_teach_options:
324
  if st.checkbox(model, value=True, key=f"checkbox_{model}"):
325
  co_teach_llms.append(model)