andito HF staff commited on
Commit
3d2e344
1 Parent(s): ce42461

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)