def tokenize_query(tokenizer, prompt):
    """ Prepare input """
    prompt = f"<|endoftext|> english: {prompt}\nbash:"
    encoded_prompt = tokenizer(prompt, return_tensors="pt")
    return encoded_prompt