Update app.py

app.py

@@ -11,6 +11,19 @@ def get_model():
     
 col1, col2 = st.columns([2,1])
 
+with st.sidebar:
+    st.markdown('## Model Parameters')
+
+    max_length = st.slider('Max text length', 0, 150, 80)
+
+    num_beams = st.slider('N° tree bams search', 2, 15,  5)
+
+    early_stopping = st.selectbox(
+     'Early stopping text generation',
+     ('True', 'False'), key={'True' : True, 'False': False}, index=0)
+
+    no_ngram_repeat = st.slider('Max repetition limit', 1, 5,  2)
+    
 with col1:
     prompt= st.text_area('Your prompt here',
         '''Who is Elon Musk?''') 
@@ -33,8 +46,9 @@ with col2:
      
 gen = st.info('Generating text...')
 answer = generator(prompt,
-                       max_length=100,
-                       no_repeat_ngram_size=3, early_stopping=True, num_beams=10, skip_special_tokens=True)                      
+                       max_length=max_length,no_repeat_ngram_size=no_ngram_repeat,
+                        early_stopping=early_stopping, num_beams=num_beams,
+                          skip_special_tokens=True)                      
 gen.empty()                      
                        
 lst = answer[0]['generated_text']