Pippoz commited on
Commit
600205e
1 Parent(s): c0b285b

Update app.py

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