from transformers import AutoModelForCausalLM, AutoTokenizerimport torch model_id = "Nitesh208/secureiq-threat-classifier"tokenizer = AutoTokenizer.from_pretrained(model_id)model = AutoModelForCausalLM.from_pretrained( model_id, torch_dtype=torch.float16, device_map="auto") prompt = """<|begin_of_text|><|start_header_id|>system<|end_header_id|>You are SecureIQ, an expert AI security analyst.<|eot_id|><|start_header_id|>user<|end_header_id|>Classify this vulnerability: Remote code execution via buffer overflow<|eot_id|><|start_header_id|>assistant<|end_header_id|>""" inputs = tokenizer(prompt, return_tensors="pt")outputs = model.generate(**inputs, max_new_tokens=150)print(tokenizer.decode(outputs[0], skip_special_tokens=True))