breadlicker45 commited on
Commit
b4fecc1
1 Parent(s): d84e31b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -31,14 +31,16 @@ with col1:
31
  with col2:
32
  select_model = st.radio(
33
  "Select the model to use:",
34
- ('MuseWeb', 'MusePy', 'MuseNeo'), index = 2)
35
 
36
  if select_model == 'MuseWeb':
37
  model = 'breadlicker45/museweb'
38
  elif select_model == 'MusePy':
39
  model = 'breadlicker45/MusePy'
40
  elif select_model == 'MuseNeo':
41
- model = 'breadlicker45/MuseNeo'
 
 
42
 
43
  with st.spinner('Loading Model... (This may take a while)'):
44
  generator = get_model()
 
31
  with col2:
32
  select_model = st.radio(
33
  "Select the model to use:",
34
+ ('MuseWeb', 'MusePy', 'MuseNeo', 'MusePy-1-1'), index = 2)
35
 
36
  if select_model == 'MuseWeb':
37
  model = 'breadlicker45/museweb'
38
  elif select_model == 'MusePy':
39
  model = 'breadlicker45/MusePy'
40
  elif select_model == 'MuseNeo':
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()