Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ def extend(input_text, num_return_sequences, bad_words, max_size=20, top_k=50, t
|
|
38 |
input_ids=input_ids,
|
39 |
max_length=max_size + len(encoded_prompt[0]),
|
40 |
top_k=top_k,
|
41 |
-
|
42 |
top_p=top_p,
|
43 |
do_sample=True,
|
44 |
num_return_sequences=num_return_sequences)
|
|
|
38 |
input_ids=input_ids,
|
39 |
max_length=max_size + len(encoded_prompt[0]),
|
40 |
top_k=top_k,
|
41 |
+
bad_words_ids = bad_word_ids,
|
42 |
top_p=top_p,
|
43 |
do_sample=True,
|
44 |
num_return_sequences=num_return_sequences)
|