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 |
|