breadlicker45 commited on
Commit
4e6f33f
1 Parent(s): d0d5a73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ with st.sidebar:
16
 
17
  max_length = st.slider('Max text length', 0, 150, 80)
18
 
19
- num_beams = st.slider('N° tree beams search', 2, 15, 5)
20
 
21
  early_stopping = st.selectbox(
22
  'Early stopping text generation',
@@ -38,7 +38,7 @@ with col2:
38
  elif select_model == 'OPT-350m':
39
  model = 'facebook/opt-350m'
40
  elif select_model == 'OPT-125m':
41
- model = 'facebook/opt-125m'
42
 
43
  with st.spinner('Loading Model... (This may take a while)'):
44
  generator = get_model()
 
16
 
17
  max_length = st.slider('Max text length', 0, 150, 80)
18
 
19
+ num_beams = st.slider('N° tree beams search', 2, 15, 0)
20
 
21
  early_stopping = st.selectbox(
22
  'Early stopping text generation',
 
38
  elif select_model == 'OPT-350m':
39
  model = 'facebook/opt-350m'
40
  elif select_model == 'OPT-125m':
41
+ model = 'BAAI/glm-515m'
42
 
43
  with st.spinner('Loading Model... (This may take a while)'):
44
  generator = get_model()