diff --git a/AITrain/model_download.py b/AITrain/model_download.py
index e3b66a2..f6d9fd0 100644
--- a/AITrain/model_download.py
+++ b/AITrain/model_download.py
@@ -1,2 +1,3 @@
 from modelscope import snapshot_download
-model_dir = snapshot_download('deepseek-ai/DeepSeek-R1-Distill-Qwen-7B', cache_dir='./', revision='master')
\ No newline at end of file
+model_dir = snapshot_download('Qwen/Qwen3-4B', cache_dir='./', revision='master')
+snapshot_download('sentence-transformers/all-MiniLM-L6-v2', cache_dir='./', revision='master')
\ No newline at end of file
diff --git a/AITrain/requirements.txt b/AITrain/requirements.txt
index 3e6f987..c717db9 100644
--- a/AITrain/requirements.txt
+++ b/AITrain/requirements.txt
@@ -16,4 +16,5 @@ datasets==3.5.1
 peft==0.15.2
 autoawq
 PyPDF2
-sentence-transformersfaiss-cpu
\ No newline at end of file
+sentence-transformersfaiss-cpu
+fitz
\ No newline at end of file