DHRUV SHEKHAWAT commited on
Commit
a667f0a
·
1 Parent(s): ff8b681

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -25,7 +25,7 @@ if 'api_key' not in st.session_state:
25
  st.session_state['api_key']=""
26
 
27
  st.sidebar.title("Settings")
28
- model_name = st.sidebar.selectbox("Model:", ("gpt3.5","gpt3.5-16k","gpt4-32k","mini-orca-v3-13b"))
29
  counter_placeholder = st.sidebar.empty()
30
 
31
  api_key = st.sidebar.text_input("API_Key", value=st.session_state['api_key'], max_chars=None,placeholder=st.session_state['api_key'], key=None, type="password", label_visibility="visible")
@@ -43,16 +43,16 @@ if 'messages' not in st.session_state:
43
  st.session_state['messages'] = [
44
  {"role": "system", "content": system_message}
45
  ]
46
- if model_name == "gpt3.5":
47
  model = "gpt-3.5-turbo"
48
- elif model_name == "gpt3.5-16k":
49
  model = "gpt-3.5-turbo-16k"
50
- elif model_name == "gpt4-32k":
51
  model = "gpt4-32k"
52
  elif model_name == "mini-orca-v3-13b":
53
  model = "orca-mini-v3-13b"
54
  else:
55
- model = "gpt-3.5-turbo"
56
 
57
  if clear_button:
58
  st.session_state['generated'] = []
 
25
  st.session_state['api_key']=""
26
 
27
  st.sidebar.title("Settings")
28
+ model_name = st.sidebar.selectbox("Model:", ("gpt-3.5-turbo","gpt-3.5-turbo-16k","gpt-4-32k","claude-1","claude-2","palm-2","codellama-7b","llama2-7b","llama2-70b","openassistant-llama-30b","mini-orca-v3-13b"))
29
  counter_placeholder = st.sidebar.empty()
30
 
31
  api_key = st.sidebar.text_input("API_Key", value=st.session_state['api_key'], max_chars=None,placeholder=st.session_state['api_key'], key=None, type="password", label_visibility="visible")
 
43
  st.session_state['messages'] = [
44
  {"role": "system", "content": system_message}
45
  ]
46
+ if model_name == "gpt-3.5-turbo":
47
  model = "gpt-3.5-turbo"
48
+ elif model_name == "gpt-3.5-turbo-16k":
49
  model = "gpt-3.5-turbo-16k"
50
+ elif model_name == "gpt-4-32k":
51
  model = "gpt4-32k"
52
  elif model_name == "mini-orca-v3-13b":
53
  model = "orca-mini-v3-13b"
54
  else:
55
+ model = model_name
56
 
57
  if clear_button:
58
  st.session_state['generated'] = []