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 OPT from  Facebook')
+st.markdown('## Text-generation gpt Muse models from Breadlicker')
 
 @st.cache(allow_output_mutation=True, suppress_st_warning =True, show_spinner=False)
 def get_model():
@@ -14,9 +14,9 @@ col1, col2 = st.columns([2,1])
 with st.sidebar:
     st.markdown('## Model Parameters')
 
-    max_length = st.slider('Max text length', 0, 250, 80)
+    max_length = st.slider('Max text length', 0, 500, 80)
 
-    num_beams = st.slider('N° tree beams search', 2, 15,  0)
+    num_beams = st.slider('N° tree beams search', 2, 15,  2)
 
     early_stopping = st.selectbox(
      'Early stopping text generation',
@@ -26,19 +26,19 @@ with st.sidebar:
     
 with col1:
     prompt= st.text_area('Your prompt here',
-        '''Who is Elon Musk?''') 
+        '''2623 2619 3970 3976 2607 3973 2735 3973 2598 3985 2726 3973 2607 4009 2735 3973 2598 3973 2726 3973 2607 3973 2735 4009''') 
         
 with col2:
     select_model = st.radio(
         "Select the model to use:",
-        ('OPT-125m', 'OPT-350m', 'OPT-1.3b'), index = 1)
+        ('MuseWeb', 'MusePy', 'MuseNeo'), index = 2)
 
-    if select_model == 'OPT-1.3b':
-        model = 'facebook/opt-1.3b'
-    elif select_model == 'OPT-350m':
+    if select_model == 'MuseWeb':
+        model = 'breadlicker45/MuseWeb'
+    elif select_model == 'MusePy':
         model = 'breadlicker45/MusePy'
-    elif select_model == 'OPT-125m':
-        model = 'BAAI/glm-515m'   
+    elif select_model == 'MuseNeo':
+        model = 'breadlicker45/MuseNeo'   
 
     with st.spinner('Loading Model... (This may take a while)'):
         generator = get_model()