breadlicker45 commited on
Commit
ab9b4d4
·
1 Parent(s): f234dcc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ with col2:
45
  gen = st.info('Generating text...')
46
  answer = generator(prompt,
47
  max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
48
- early_stopping=early_stopping, num_beams=num_beams, do_sample=False, bad_words_ids="\n")
49
  gen.empty()
50
 
51
  lst = answer[0]['generated_text']
 
45
  gen = st.info('Generating text...')
46
  answer = generator(prompt,
47
  max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
48
+ early_stopping=early_stopping, num_beams=num_beams, do_sample=False)
49
  gen.empty()
50
 
51
  lst = answer[0]['generated_text']