breadlicker45 commited on
Commit
5b872f6
1 Parent(s): 9dc6d1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -31,7 +31,7 @@ with col1:
31
  with col2:
32
  select_model = st.radio(
33
  "Select the model to use:",
34
- ('MuseWeb', 'MusePy', 'MuseNeo', 'MusePy-1-1'), index = 3)
35
 
36
  if select_model == 'MuseWeb':
37
  model = 'breadlicker45/museweb'
@@ -41,6 +41,8 @@ with col2:
41
  model = 'breadlicker45/MuseNeo'
42
  elif select_model == 'MusePy-1-1':
43
  model = 'BreadAi/MusePy-1-1'
 
 
44
 
45
  with st.spinner('Loading Model... (This may take a while)'):
46
  generator = get_model()
 
31
  with col2:
32
  select_model = st.radio(
33
  "Select the model to use:",
34
+ ('MuseWeb', 'MusePy', 'MuseNeo', 'MusePy-1-1', 'MuseCan (very bad)'), index = 3)
35
 
36
  if select_model == 'MuseWeb':
37
  model = 'breadlicker45/museweb'
 
41
  model = 'breadlicker45/MuseNeo'
42
  elif select_model == 'MusePy-1-1':
43
  model = 'BreadAi/MusePy-1-1'
44
+ elif select_model == 'MuseCan (very bad)':
45
+ model = 'BreadAi/MuseCan'
46
 
47
  with st.spinner('Loading Model... (This may take a while)'):
48
  generator = get_model()