import torchfrom transformers import AutoTokenizer, AutoModelForCausalLMfrom peft import PeftModel base = AutoModelForCausalLM.from_pretrained( "Qwen/Qwen3-1.7B", torch_dtype=torch.bfloat16, low_cpu_mem_usage=True)model = PeftModel.from_pretrained(base, "jahnavidanda02/drivesignal-qwen3-lora")tokenizer = AutoTokenizer.from_pretrained("jahnavidanda02/drivesignal-qwen3-lora") taxonomy = ["Perception Failure", "Prediction Failure", "Lane Keeping", "Braking Behavior", "Unwanted Maneuver", "Construction/Environment", "Precautionary", "System/Hardware Fault", "Localization/Mapping", "Other"]prompt = (f"Classify this AV disengagement into one category from: {', '.join(taxonomy)}.\n\n" "Description: Vehicle disengaged after hesitating at an unprotected left turn.\n\nCategory:")inputs = tokenizer(prompt, return_tensors="pt")out = model.generate(**inputs, max_new_tokens=15, do_sample=False)print(tokenizer.decode(out[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))