aquibmoin commited on
Commit
eb7ca70
1 Parent(s): 9193677

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,9 +22,9 @@ with st.sidebar:
22
  st.subheader('Models and parameters')
23
  selected_model = st.sidebar.selectbox('Choose a Llama2 model', ['Llama2-7B', 'Llama2-13B'], key='selected_model')
24
  if selected_model == 'Llama2-7B':
25
- llm = 'a16z-infra/llama7b-v2-chat:4f0a4744c7295c024a1de15e1a63c880d3da035fa1f49bfd344fe076074c8eea'
26
  elif selected_model == 'Llama2-13B':
27
- llm = 'a16z-infra/llama13b-v2-chat:df7690f1994d94e96ad9d568eac121aecf50684a0b0963b25a41cc40061269e5'
28
  temperature = st.sidebar.slider('temperature', min_value=0.01, max_value=5.0, value=0.1, step=0.01)
29
  top_p = st.sidebar.slider('top_p', min_value=0.01, max_value=1.0, value=0.9, step=0.01)
30
  max_length = st.sidebar.slider('max_length', min_value=32, max_value=1000, value=400, step=8)
 
22
  st.subheader('Models and parameters')
23
  selected_model = st.sidebar.selectbox('Choose a Llama2 model', ['Llama2-7B', 'Llama2-13B'], key='selected_model')
24
  if selected_model == 'Llama2-7B':
25
+ llm = 'meta-llama/Llama-2-7b-chat'
26
  elif selected_model == 'Llama2-13B':
27
+ llm = 'meta-llama/Llama-2-13b-chat'
28
  temperature = st.sidebar.slider('temperature', min_value=0.01, max_value=5.0, value=0.1, step=0.01)
29
  top_p = st.sidebar.slider('top_p', min_value=0.01, max_value=1.0, value=0.9, step=0.01)
30
  max_length = st.sidebar.slider('max_length', min_value=32, max_value=1000, value=400, step=8)