from transformers import AutoModelForCausalLM, AutoTokenizerimport torch model_id = "vanshkamra12/CyberSecurity-Model" tokenizer = AutoTokenizer.from_pretrained(model_id)model = AutoModelForCausalLM.from_pretrained( model_id, torch_dtype=torch.float16, device_map="auto") prompt = """Below is an instruction that describes a task, paired with an input that provides further context. Write a response that appropriately completes the request. ### Instruction:Analyze the following vulnerability data and produce a structured threat intelligence report. ### Input:CVE ID: CVE-2024-21762Description: A out-of-bound write vulnerability in FortiOS SSL VPN allows a remote unauthenticated attacker to execute arbitrary code or commands via specially crafted HTTP requests.CVSS Score: 9.8 CRITICALVendor: Fortinet ### Response:""" inputs = tokenizer(prompt, return_tensors="pt").to(model.device)outputs = model.generate(**inputs, max_new_tokens=1000, temperature=0.7)print(tokenizer.decode(outputs[0], skip_special_tokens=True))