Update app.py

app.py

@@ -26,7 +26,7 @@ with st.sidebar:
      'Early stopping text generation',
      ('True', 'False'), key={'True' : True, 'False': False}, index=0)
 
-    no_ngram_repeat = st.slider('Max repetition limit', 1, 3,  1)
+    no_ngram_repeat = st.slider('Max repetition limit', 1, 3,  2)
 
     st.markdown('## how to convert it into midi. go to this site https://mrcheeze.github.io/musenet-midi/ and then paste the numbers/musenet encoders you get from the ai into the big box and then click export midi')
 
@@ -38,7 +38,7 @@ with col1:
 with col2:
     select_model = st.radio(
         "Select the model to use:",
-        ('MuseWeb', 'MusePy', 'MuseNeo', 'MusePy-1-1', 'MuseCan'), index = 4)
+        ('MuseWeb', 'MusePy', 'MuseNeo', 'MusePy-1-1', 'MuseCan', 'MuseCan-1-1'), index = 5)
 
     if select_model == 'MuseWeb':
         model = 'breadlicker45/museweb'
@@ -50,6 +50,8 @@ with col2:
         model = 'BreadAi/MusePy-1-1'
     elif select_model == 'MuseCan':
         model = 'BreadAi/MuseCan'
+    elif select_model == 'MuseCan-1-1':
+        model = 'BreadAi/MuseCan-1-1'
 
     with st.spinner('Loading Model... (This may take a while)'):
         generator = get_model()