matthewgrossman pmlocek commited on
Commit
6a77fdf
1 Parent(s): faef657

Update models. (#2)

Browse files

- Update models. (8983b86a3a15779f7de6d715c586c73d741c1154)


Co-authored-by: Piotr Mlocek <[email protected]>

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)