Update app.py

app.py

@@ -3,7 +3,7 @@ import time
 from transformers import pipeline
 import torch
 trust_remote_code=True
-st.markdown('## Text-generation gpt Muse from Breadlicker45')
+st.markdown('## story-generation from Breadlicker45')
 use_auth_token=True
 @st.cache(allow_output_mutation=True, suppress_st_warning =True, show_spinner=False)
 def get_model():
@@ -28,22 +28,21 @@ with st.sidebar:
 
     no_ngram_repeat = st.slider('Max repetition limit', 1, 3,  1)
 
-    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')
 
     
 with col1:
     prompt= st.text_area('Your prompt here',
-        '''2623 2619 3970 3976 2607 3973 2735 3973 2598 3985 2726 3973 2607 4009 2735 3973 2598 3973 2726 3973 2607 3973 2735 4009''') 
+        '''in a world''') 
         
 with col2:
     select_model = st.radio(
         "Select the model to use:",
-        ('MuseWeb', 'MusePy', 'MuseNeo', 'MusePy-1-1', 'MuseCan'), index = 4)
+        ('StoryPy', 'null'), index = 0)
 
-    if select_model == 'MuseWeb':
-        model = 'breadlicker45/museweb'
-    elif select_model == 'MusePy':
-        model = 'breadlicker45/MusePy'
+    if select_model == 'StoryPy':
+        model = 'BreadAi/StoryPy'
+    elif select_model == 'null':
+        model = 'BreadAi/StoryPy'
     elif select_model == 'MuseNeo':
         model = 'breadlicker45/MuseNeo'
     elif select_model == 'MusePy-1-1':