lfoppiano commited on
Commit
7d19a12
1 Parent(s): ab9a153

replace combo with dropdown box

Browse files
Files changed (1) hide show
  1. streamlit_app.py +10 -10
streamlit_app.py CHANGED
@@ -206,17 +206,17 @@ def play_old_messages():
206
  # is_api_key_provided = st.session_state['api_key']
207
 
208
  with st.sidebar:
209
- st.session_state['model'] = model = st.radio(
210
- "Model",
211
- ("chatgpt-3.5-turbo", "mistral-7b-instruct-v0.1", "zephyr-7b-beta"),
 
 
 
212
  index=2,
213
- captions=[
214
- "ChatGPT 3.5 Turbo + Ada-002-text (embeddings)",
215
- "Mistral-7B-Instruct-V0.1 + Sentence BERT (embeddings) :free:",
216
- "Zephyr-7B-beta + Sentence BERT (embeddings) :free:"
217
- ],
218
- help="Select the LLM model and embeddings you want to use.",
219
- disabled=st.session_state['doc_id'] is not None or st.session_state['uploaded'])
220
 
221
  st.markdown(
222
  ":warning: Mistral and Zephyr are **FREE** to use. Requests might fail anytime. Use at your own risk. :warning: ")
 
206
  # is_api_key_provided = st.session_state['api_key']
207
 
208
  with st.sidebar:
209
+ st.session_state['model'] = model = st.selectbox(
210
+ "Model:",
211
+ options={
212
+ "chatgpt-3.5-turbo": "ChatGPT 3.5 Turbo + Ada-002-text (embeddings)",
213
+ "mistral-7b-instruct-v0.1": "Mistral-7B-Instruct-V0.1 + Sentence BERT (embeddings) :free:",
214
+ "zephyr-7b-beta": "Zephyr-7B-beta + Sentence BERT (embeddings) :free:"},
215
  index=2,
216
+ placeholder="Select model",
217
+ help="Select the LLM model:",
218
+ disabled=st.session_state['doc_id'] is not None or st.session_state['uploaded']
219
+ )
 
 
 
220
 
221
  st.markdown(
222
  ":warning: Mistral and Zephyr are **FREE** to use. Requests might fail anytime. Use at your own risk. :warning: ")