Amironox commited on
Commit
2031fdd
·
verified ·
1 Parent(s): fb2cbf4

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +3 -1
src/streamlit_app.py CHANGED
@@ -36,11 +36,13 @@ else:
36
  api_key = st.session_state.api_key
37
  base_url = endpoint
38
  client = OpenAI(api_key=api_key, base_url=base_url)
39
-
40
  # Extract the model name
41
  models = client.models.list()
42
  modelname = models.data[0].id
43
 
 
 
44
  if "messages" not in st.session_state:
45
  st.session_state.messages = []
46
 
 
36
  api_key = st.session_state.api_key
37
  base_url = endpoint
38
  client = OpenAI(api_key=api_key, base_url=base_url)
39
+
40
  # Extract the model name
41
  models = client.models.list()
42
  modelname = models.data[0].id
43
 
44
+ st.write(f"**Model Info:** `{modelname}`")
45
+
46
  if "messages" not in st.session_state:
47
  st.session_state.messages = []
48