breadlicker45 commited on
Commit
dc0567c
1 Parent(s): 6df1751

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -40,9 +40,7 @@ with col2:
40
  "Select the model to use:",
41
  ('MusePy', 'MuseNeo', 'MusePy-1-1', 'MuseCan', 'MuseCan-1-1'), index = 4)
42
 
43
- if select_model == 'MuseCan-1-1':
44
- model = 'BreadAi/MuseCan-1-1'
45
- elif select_model == 'MusePy':
46
  model = 'breadlicker45/MusePy'
47
  elif select_model == 'MuseNeo':
48
  model = 'breadlicker45/MuseNeo'
@@ -50,6 +48,8 @@ with col2:
50
  model = 'BreadAi/MusePy-1-1'
51
  elif select_model == 'MuseCan':
52
  model = 'BreadAi/MuseCan'
 
 
53
 
54
  with st.spinner('Loading Model... (This may take a while)'):
55
  generator = get_model()
 
40
  "Select the model to use:",
41
  ('MusePy', 'MuseNeo', 'MusePy-1-1', 'MuseCan', 'MuseCan-1-1'), index = 4)
42
 
43
+ if select_model == 'MusePy':
 
 
44
  model = 'breadlicker45/MusePy'
45
  elif select_model == 'MuseNeo':
46
  model = 'breadlicker45/MuseNeo'
 
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()