VISION-1 is a fine-tuned version of Llama 3.1 8B Instruct, specialized for content safety analysis and moderation. The model is trained to identify and analyze potential safety concerns in text content, including scams, fraud, harmful content, and inappropriate material.
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3# Load model and tokenizer
4tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.1-8B-Instruct")
5model = AutoModelForCausalLM.from_pretrained("OverseerAI/VISION-1")
6
7# Format prompt
8prompt = "Analyze the following content for safety concerns: 'Click here to win a free iPhone! Just enter your credit card details.'"
9formatted_prompt = f"<s>[INST] {prompt} [/INST]"
10
11# Generate response
12inputs = tokenizer(formatted_prompt, return_tensors="pt", padding=True)
13outputs = model.generate(**inputs, max_new_tokens=128)
14response = tokenizer.decode(outputs[0], skip_special_tokens=True)