breadlicker45 commited on
Commit
f512a02
1 Parent(s): eb5f7d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -31,11 +31,11 @@ with col1:
31
  with col2:
32
  select_model = st.radio(
33
  "Select the model to use:",
34
- ('MuseWeb', 'gpt-YA-1-1'), index = 1)
35
 
36
- if select_model == 'MuseWeb':
37
  model = 'breadlicker45/gpt-ya'
38
- elif select_model == 'gpt-YA-1-1':
39
  model = 'BreadAi/gpt-YA-1-1_70M'
40
 
41
  with st.spinner('Loading Model... (This may take a while)'):
 
31
  with col2:
32
  select_model = st.radio(
33
  "Select the model to use:",
34
+ ('gpt-ya', 'gpt-ya-1-1'), index = 1)
35
 
36
+ if select_model == 'gpt-ya':
37
  model = 'breadlicker45/gpt-ya'
38
+ elif select_model == 'gpt-ya-1-1':
39
  model = 'BreadAi/gpt-YA-1-1_70M'
40
 
41
  with st.spinner('Loading Model... (This may take a while)'):