Pippoz commited on
Commit
6b59e75
1 Parent(s): 9e2e1de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -31,17 +31,14 @@ with col1:
31
  with col2:
32
  select_model = st.radio(
33
  "Select the model to use:",
34
- ('OPT-125m', 'OPT-350m', 'OPT-1.3b', 'OPT-2.7b'), index = 1)
35
 
36
  if select_model == 'OPT-1.3b':
37
  model = 'facebook/opt-1.3b'
38
  elif select_model == 'OPT-350m':
39
  model = 'facebook/opt-350m'
40
  elif select_model == 'OPT-125m':
41
- model = 'facebook/opt-125m'
42
- elif select_model == 'OPT-2.7b':
43
- model = 'facebook/opt-2.7b'
44
-
45
 
46
  with st.spinner('Loading Model... (This may take a while)'):
47
  generator = get_model()
 
31
  with col2:
32
  select_model = st.radio(
33
  "Select the model to use:",
34
+ ('OPT-125m', 'OPT-350m', 'OPT-1.3b'), index = 1)
35
 
36
  if select_model == 'OPT-1.3b':
37
  model = 'facebook/opt-1.3b'
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()