from transformers import AutoModelForCausalLM, AutoTokenizerĀ model_name = "zenlm/zen-guard-gen"Ā tokenizer = AutoTokenizer.from_pretrained(model_name)model = AutoModelForCausalLM.from_pretrained( model_name, torch_dtype="auto", device_map="auto")Ā # Prompt moderationprompt = "How do I learn programming?"messages = [{"role": "user", "content": prompt}]text = tokenizer.apply_chat_template(messages, tokenize=False)inputs = tokenizer([text], return_tensors="pt").to(model.device)Ā outputs = model.generate(**inputs, max_new_tokens=128)result = tokenizer.decode(outputs[0][len(inputs.input_ids[0]):], skip_special_tokens=True)print(result)# Output: Safety: Safe# Categories: NoneĀ # Response moderationresponse = "Here's a Python tutorial..."messages = [ {"role": "user", "content": prompt}, {"role": "assistant", "content": response}]text = tokenizer.apply_chat_template(messages, tokenize=False)inputs = tokenizer([text], return_tensors="pt").to(model.device)outputs = model.generate(**inputs, max_new_tokens=128)result = tokenizer.decode(outputs[0][len(inputs.input_ids[0]):], skip_special_tokens=True)print(result)# Output: Safety: Safe# Categories: None# Refusal: No