Update app.py

app.py

@@ -20,6 +20,8 @@ with st.sidebar:
 
     num_beams = st.slider('N° tree beams search', 1, 15,  1)
 
+    top_k = st.slider('top_k', 1, 10,  1)
+
     temperature = st.slider('temperature', 0.0, 1.0, 0.5, 0.1)
     
     early_stopping = st.selectbox(
@@ -56,9 +58,7 @@ with col2:
         st.success('Model loaded correctly!')
      
 gen = st.info('Generating text...')
-answer = generator(prompt,
-                       max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
-                        early_stopping=early_stopping, num_beams=num_beams, min_length=min_length, temperature=temperature)                      
+answer = generator(prompt,max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,early_stopping=early_stopping, num_beams=num_beams, min_length=min_length, temperature=temperature, top_k=top_k)                      
 gen.empty()                      
                        
 lst = answer[0]['generated_text']