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

Update app.py

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