add input check

prompt_check.py

@@ -32,4 +32,6 @@ def is_unsafe_prompt(model, tokenizer, user_prompt):
 
     content = tokenizer.decode(output_ids, skip_special_tokens=True).strip("\n")
 
+    print(content)
+
     return "yes" in content.lower()