breadlicker45 commited on
Commit
af8c375
1 Parent(s): f75fd35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -38,7 +38,7 @@ with col1:
38
  with col2:
39
  select_model = st.radio(
40
  "Select the model to use:",
41
- ('MusePy', 'MuseMini', 'MusePy-1-1', 'MuseCan', 'MuseCan-1-1'), index = 4)
42
 
43
  if select_model == 'MusePy':
44
  model = 'breadlicker45/MusePy'
@@ -48,8 +48,8 @@ with col2:
48
  model = 'BreadAi/MusePy-1-1'
49
  elif select_model == 'MuseCan':
50
  model = 'BreadAi/MuseCan'
51
- elif select_model == 'MuseCan-1-1':
52
- model = 'BreadAi/MuseCan-1-1'
53
 
54
  with st.spinner('Loading Model... (This may take a while)'):
55
  generator = get_model()
 
38
  with col2:
39
  select_model = st.radio(
40
  "Select the model to use:",
41
+ ('MusePy', 'MuseMini', 'MusePy-1-1', 'MuseCan', 'MuseCan-1-2'), index = 4)
42
 
43
  if select_model == 'MusePy':
44
  model = 'breadlicker45/MusePy'
 
48
  model = 'BreadAi/MusePy-1-1'
49
  elif select_model == 'MuseCan':
50
  model = 'BreadAi/MuseCan'
51
+ elif select_model == 'MuseCan-1-2':
52
+ model = 'BreadAi/MuseCan-1-2'
53
 
54
  with st.spinner('Loading Model... (This may take a while)'):
55
  generator = get_model()