breadlicker45 commited on
Commit
5621e21
1 Parent(s): dcf7790

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ with col2:
49
  gen = st.info('Generating text...')
50
  answer = generator(prompt,
51
  max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
52
- early_stopping=early_stopping, num_beams=num_beams, do_sample=True)
53
  gen.empty()
54
 
55
  lst = answer[0]['generated_text']
 
49
  gen = st.info('Generating text...')
50
  answer = generator(prompt,
51
  max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
52
+ early_stopping=early_stopping, num_beams=num_beams, do_sample=False)
53
  gen.empty()
54
 
55
  lst = answer[0]['generated_text']