Update app.py
Browse files
app.py
CHANGED
@@ -67,6 +67,7 @@ def model_inference(
|
|
67 |
generation_args["temperature"] = temperature
|
68 |
generation_args["do_sample"] = True
|
69 |
generation_args["top_p"] = top_p
|
|
|
70 |
|
71 |
|
72 |
generation_args.update(inputs)
|
|
|
67 |
generation_args["temperature"] = temperature
|
68 |
generation_args["do_sample"] = True
|
69 |
generation_args["top_p"] = top_p
|
70 |
+
generation_args["bad_words_ids"] = BAD_WORDS_IDS
|
71 |
|
72 |
|
73 |
generation_args.update(inputs)
|