from transformers import AutoTokenizer, AutoModelForCausalLMfrom peft import PeftModelimport torch base = "Qwen/Qwen2.5-7B-Instruct"lora_repo = "Ushitha/ushitha-coder-network-corrector" tokenizer = AutoTokenizer.from_pretrained(base)model = AutoModelForCausalLM.from_pretrained( base, torch_dtype=torch.bfloat16, device_map="auto")model = PeftModel.from_pretrained(model, lora_repo) insecure_config = """hostname Routerinterface GigabitEthernet0/0 ip address 192.168.1.1 255.255.255.0 no shutdown""" messages = [ {"role": "system", "content": "You are a network security expert..."}, {"role": "user", "content": f"Review this config:\n{insecure_config}"},]text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)inputs = tokenizer(text, return_tensors="pt").to(model.device)output = model.generate(**inputs, max_new_tokens=2048, temperature=0.1, do_sample=True)print(tokenizer.decode(output[0], skip_special_tokens=True))