Update app.py

app.py

@@ -14,7 +14,7 @@ col1, col2 = st.columns([2,1])
 with st.sidebar:
     st.markdown('## Model Parameters')
 
-    max_length = st.slider('Max text length', 0, 2000, 80)
+    max_length = st.slider('Max text length', 0, 500, 80)
 
     num_beams = st.slider('N° tree beams search', 1, 15,  1)
 
@@ -31,16 +31,12 @@ with col1:
 with col2:
     select_model = st.radio(
         "Select the model to use:",
-        ('MuseWeb', 'MusePy', 'MuseNeo', 'MusePy-1-1'), index = 3)
+        ('MuseWeb', 'gpt-YA-1-1'), index = 1)
 
     if select_model == 'MuseWeb':
-        model = 'breadlicker45/museweb'
-    elif select_model == 'MusePy':
-        model = 'breadlicker45/MusePy'
-    elif select_model == 'MuseNeo':
-        model = 'breadlicker45/MuseNeo'
-    elif select_model == 'MusePy-1-1':
-        model = 'BreadAi/MusePy-1-1'
+        model = 'breadlicker45/gpt-ya'
+    elif select_model == 'gpt-YA-1-1':
+        model = 'BreadAi/gpt-YA-1-1_70M'
 
     with st.spinner('Loading Model... (This may take a while)'):
         generator = get_model()