Benjamin Consolvo
commited on
Commit
·
caad47f
1
Parent(s):
fa5b702
model.id for model names list
Browse files
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 |
-
|
|
|
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 |
|