This commit is contained in:
Xi Yan 2024-12-02 13:06:36 -08:00
parent 9bceb1912e
commit 7f2ed9622c
4 changed files with 8 additions and 12 deletions

View file

@ -10,7 +10,7 @@ from modules.api import llama_stack_api
# Sidebar configurations
with st.sidebar:
st.header("Configuration")
available_models = llama_stack_api.list_models()
available_models = llama_stack_api.client.models.list()
available_models = [model.identifier for model in available_models]
selected_model = st.selectbox(
"Choose a model",