Update app.py

app.py

@@ -22,7 +22,7 @@ with st.sidebar:
      'Early stopping text generation',
      ('True', 'False'), key={'True' : True, 'False': False}, index=0)
 
-    no_ngram_repeat = st.slider('Max repetition limit', 10, 30,  15)
+    no_ngram_repeat = st.slider('Max repetition limit', 1, 3,  2)
     
 with col1:
     prompt= st.text_area('Your prompt here',
@@ -48,8 +48,8 @@ with col2:
      
 gen = st.info('Generating text...')
 answer = generator(prompt,
-                       max_length=max_length, no_repeat_ngram_size=(no_ngram_repeat/10),
-                        early_stopping=early_stopping, num_beams=num_beams)                      
+                       max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
+                        early_stopping=early_stopping, num_beams=num_beams, do_sample=True)                      
 gen.empty()                      
                        
 lst = answer[0]['generated_text']