Update app.py

app.py

@@ -44,11 +44,10 @@ with col2:
      
 gen = st.info('Generating text...')
 bad_words_ids = [
-    tokenizer.encode(bad_word, add_prefix_space=True) for bad_word in ["\n"]
+    generator.encode(bad_word, add_prefix_space=True) for bad_word in ["\n"]
 ]
-answer = generator(prompt,
-                       max_length=max_length, no_repeat_ngram_size=no_ngram_repeat,
-                        early_stopping=early_stopping, num_beams=num_beams, do_sample=False, bad_words_ids=bad_words_ids)                      
+answer = generator(prompt, max_length=max_length, no_repeat_ngram_size=no_ngram_repeat, 
+                   early_stopping=early_stopping, num_beams=num_beams, do_sample=False, bad_words_ids=bad_words_ids)                      
 gen.empty()                      
                        
 lst = answer[0]['generated_text']