diff --git a/README.md b/README.md
index 46d2a20..b9fe78a 100644
--- a/README.md
+++ b/README.md
@@ -2,6 +2,9 @@
接口文档:http://127.0.0.1:8000/docs。
+
+AI模型的本地地址 存储路径:/Users/bmy/.cache/modelscope/hub/damo
+
fairseq 依赖 需要 windows 先安装 vs_BuildTools.exe
decord 用 eva-decord 替代
diff --git a/requirements.txt b/requirements.txt
index 7100cf0..5995093 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,7 @@ uvicorn==0.27.1
pydantic==2.6.1
pydantic-settings==2.2.0
requests==2.31.0
-
+numpy==1.26.4
torch==2.2.0
opencv-python==4.9.0.80
open-clip-torch==2.24.0
diff --git a/src/ai/ai_test.py b/src/ai/ai_test.py
index ecac21e..6efe8b4 100644
--- a/src/ai/ai_test.py
+++ b/src/ai/ai_test.py
@@ -4,7 +4,7 @@ from modelscope.pipelines import pipeline
from modelscope.preprocessors.image import load_image
pipeline = pipeline(task=Tasks.multi_modal_embedding, model='damo/multi-modal_clip-vit-large-patch14_336_zh', model_revision='v1.0.1', device='cpu')
-input_img = load_image('https://p1.a.yximgs.com/upic/2023/12/27/13/BMjAyMzEyMjcxMzAwMzdfMjkwOTEwNDc5N18xMjA2NTYyODIyNDRfMl8z_Be33b730fcb8aaf000ba5e58bf8c3681e.jpg?tag=1-1708460644-unknown-0-huswbkiuiq-0ac88680ace16101&clientCacheKey=3x9ghy3dj8y2z5a.jpg&di=70208a94&bp=14764') # 支持皮卡丘示例图片路径/本地图片 返回PIL.Image
+input_img = load_image('https://p2.a.yximgs.com/upic/2023/10/29/18/BMjAyMzEwMjkxODA4NTRfMTUxNDU3NTMwNV8xMTYwNDQ3MzU3OThfMV8z_B903dcf4acbd7359157de400894a486b2.jpg?tag=1-1737455669-unknown-0-eevfhc2ows-f3b84d4c8f01cc99&clientCacheKey=3x49fsxn85ijp6q.jpg&di=JA4DYiIS6gDJR6IE5Fh9Mg==&bp=14764') # 支持皮卡丘示例图片路径/本地图片 返回PIL.Image
input_texts = ["小火龙", "白色衬衫"]
# 支持一张图片(PIL.Image)或多张图片(List[PIL.Image])输入,输出归一化特征向量
diff --git a/src/ai/multimModaImageTextSearch.py b/src/ai/multimModaImageTextSearch.py
index 8a7ede6..90d978b 100644
--- a/src/ai/multimModaImageTextSearch.py
+++ b/src/ai/multimModaImageTextSearch.py
@@ -5,7 +5,7 @@ from modelscope.preprocessors.image import load_image
from fastapi import WebSocket
class multimModaImageTextSearch:
- # AI模型的本地地址
+ # AI模型的本地地址 存储路径:/Users/bmy/.cache/modelscope/hub/damo
modelName: str = 'damo/multi-modal_clip-vit-large-patch14_336_zh'
# 使用 cpu 还是 GPU 计算,GPU 比 CPU快,但是 GPU 只支持 英伟达 显卡,至少4G显存以上
deviceType: str = 'cpu' # cpu, gpu
diff --git a/src/celery/ai_image/task.py b/src/celery/ai_image/task.py
index 014afaf..c1da74e 100644
--- a/src/celery/ai_image/task.py
+++ b/src/celery/ai_image/task.py
@@ -27,6 +27,7 @@ def start_ai_img(self, keyWord: str, imgList: list[str]):
key=keyWord,
imgList=imgList
)
+ print("imgList: ", imgList)
res = image.start()
return res
diff --git a/src/http/_config.py b/src/http/_config.py
index e897b59..78d38cf 100644
--- a/src/http/_config.py
+++ b/src/http/_config.py
@@ -6,7 +6,7 @@ class HttpConfig:
'Host': 'www.kuaishou.com',
'Referer': 'https://www.kuaishou.com/profile/3x2347wkfukzx4g',
'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36',
- 'Cookie': 'kpf=PC_WEB; clientid=3; did=web_fca770657d0a24abb53ddc92bb42e612; didv=1700407978000; userId=58344290; kuaishou.server.web_st=ChZrdWFpc2hvdS5zZXJ2ZXIud2ViLnN0EqABFs7urbq3vvcrvB-IRBSKHXfdVRzblSFx7udepTzeAtgCompa6-aZiv1f_Bb66SubMkCbLODJQ3cn_GxlWWWbPgbQEyul9f7SDKsGphyvK1Pk3aj-yn899D9gR-xj1wiqT_0nA9x3XcPmvuJ8qQzIeLl3236npeMV0i9uYEAeIHDp9RtvPlQ4o0piktgGjMZH13Uhk7-ZfqDuUPQ-mEH7axoSS2UWrTnDahVDKzRYjzjLpJM-IiDXKnEX31rxM9oVovhBLMPSocxSvtN7wyo1oS1nNdvjSCgFMAE; kuaishou.server.web_ph=5b6a8c20ee92347d220cbc98f705f92042fb; kpn=KUAISHOU_VISION'
+ 'Cookie': 'kpf=PC_WEB; clientid=3; did=web_1debac63b767ac564381b3e54dc1de14; userId=58344290; kuaishou.server.webday7_st=ChprdWFpc2hvdS5zZXJ2ZXIud2ViZGF5Ny5zdBKwAcl8si24RTFngM_Km0UdhjDwUfscFSn1oUV5P7SNRrD_XcCr2kca2dnV8IhO9Pn_AJ5hCLjDzEnK-2c4NKzFPfTghZop4VhZh_Xads52aoOZKXvdPJ5UIdOWkpI3IUpKKEj_skdwHAYczNqCgJA40-KFLjmF6PGUcH9F924l68rOy_rVDu912xf5zrm3V_2vXzv61lg9b2Y55uAuVm_GAJr8wO91scf8FjB3FWiz2zU-GhLcWLZU0wMn7F5ONAeuFgyQkCwiIF1D6aMj4ItLQrCzkiRZXt3U4o0QfkxNv8s9PzXJ3YGtKAUwAQ; kuaishou.server.webday7_ph=2ec00e134011cd2f1ecc4904954410a101be; kpn=KUAISHOU_VISION'
}
visionProfileLikePhotoList: str = ''' fragment photoContent on PhotoEntity {
__typename
diff --git a/test/index.html b/test/index.html
index 5a8d985..2b9ab56 100644
--- a/test/index.html
+++ b/test/index.html
@@ -41,7 +41,7 @@