Benjamin Consolvo commited on
Commit
caad47f
·
1 Parent(s): fa5b702

model.id for model names list

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -24,9 +24,10 @@ with st.sidebar:
24
  api_key = st.session_state.api_key = st.secrets["openai_apikey"]
25
  base_url = st.session_state.base_url = os.environ.get("base_url")
26
  client = OpenAI(api_key=api_key, base_url=base_url)
27
- model_names = client.models.list()
 
28
 
29
- modelname = st.selectbox("Select LLM model (Running on Intel® Gaudi®) ", model_names)
30
  st.write(f"You selected: {modelname}")
31
  st.button("Start New Chat", on_click=clear_chat)
32
 
 
24
  api_key = st.session_state.api_key = st.secrets["openai_apikey"]
25
  base_url = st.session_state.base_url = os.environ.get("base_url")
26
  client = OpenAI(api_key=api_key, base_url=base_url)
27
+ models = client.models.list()
28
+ model_names = [model.id for model in models] # Extract 'id' from each model object
29
 
30
+ modelname = st.selectbox("Select LLM model (Running on Intel® Gaudi®) on Denvr Dataworks", model_names)
31
  st.write(f"You selected: {modelname}")
32
  st.button("Start New Chat", on_click=clear_chat)
33