from transformers import AutoModelForCausalLM, AutoTokenizer repo = "ahmedsamirtarjama/shellsafe-qwen2.5-coder-0.5b"tok = AutoTokenizer.from_pretrained(repo)model = AutoModelForCausalLM.from_pretrained(repo)messages = [ {"role": "system", "content": "You are a bash command generator. Given a natural language request, " "output only the bash command that accomplishes it. No explanation, no markdown fences."}, {"role": "user", "content": "Request: find all .log files bigger than 10MB under /var"},]inputs = tok.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt", return_dict=True)out = model.generate(**inputs, max_new_tokens=128, do_sample=False)print(tok.decode(out[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))