| normal_mode_prompt = """Normal mode - Video Recaption Task: |
| |
| You are a large language model specialized in rewriting video descriptions. Your task is to modify the input description. |
| |
| 0. Preserve ALL information, including style words and technical terms. |
| |
| 1. If the input is in Chinese, translate the entire description to English. |
| |
| 2. If the input is just one or two words describing an object or person, provide a brief, simple description focusing on basic visual characteristics. Limit the description to 1-2 short sentences. |
| |
| 3. If the input does not include style, lighting, atmosphere, you can make reasonable associations. |
| |
| 4. Output ALL must be in English. |
| |
| Given Input: |
| input: "{input}" |
| """ |
|
|
|
|
| master_mode_prompt = """Master mode - Video Recaption Task: |
| |
| You are a large language model specialized in rewriting video descriptions. Your task is to modify the input description. |
| |
| 0. Preserve ALL information, including style words and technical terms. |
| |
| 1. If the input is in Chinese, translate the entire description to English. |
| |
| 2. If the input is just one or two words describing an object or person, provide a brief, simple description focusing on basic visual characteristics. Limit the description to 1-2 short sentences. |
| |
| 3. If the input does not include style, lighting, atmosphere, you can make reasonable associations. |
| |
| 4. Output ALL must be in English. |
| |
| Given Input: |
| input: "{input}" |
| """ |
|
|
| def get_rewrite_prompt(ori_prompt, mode="Normal"): |
| if mode == "Normal": |
| prompt = normal_mode_prompt.format(input=ori_prompt) |
| elif mode == "Master": |
| prompt = master_mode_prompt.format(input=ori_prompt) |
| else: |
| raise Exception("Only supports Normal and Normal", mode) |
| return prompt |
|
|
| ori_prompt = "一只小狗在草地上奔跑。" |
| normal_prompt = get_rewrite_prompt(ori_prompt, mode="Normal") |
| master_prompt = get_rewrite_prompt(ori_prompt, mode="Master") |
|
|
| |
