diff --git a/README.md b/README.md
index 9a954cd..55d3385 100644
--- a/README.md
+++ b/README.md
@@ -100,7 +100,7 @@ of the **CogVideoX** open-source model.
 This folder contains some tools for model conversion / caption generation, etc.
 
 + [convert_weight_sat2hf](tools/convert_weight_sat2hf.py): Convert SAT model weights to Huggingface model weights.
-+ [caption_demo](tools/caption_demo.py): Caption tool, a model that understands videos and outputs them in text.
++ [caption_demo](tools/caption): Caption tool, a model that understands videos and outputs them in text.
 
 ## Project Plan
 
diff --git a/README_zh.md b/README_zh.md
index 32cb615..21531a3 100644
--- a/README_zh.md
+++ b/README_zh.md
@@ -92,7 +92,7 @@ CogVideoX是 [清影](https://chatglm.cn/video) 同源的开源版本视频生
 本文件夹包含了一些工具，用于模型的转换 / Caption 等工作。
 
 + [convert_weight_sat2hf](tools/convert_weight_sat2hf.py): 将 SAT 模型权重转换为 Huggingface 模型权重。
-+ [caption_demo](tools/caption_demo.py): Caption 工具，对视频理解并用文字输出的模型。
++ [caption_demo](tools/caption): Caption 工具，对视频理解并用文字输出的模型。
 
 ## 项目规划