aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat（新功能）:

Browse Source 
fix（修复bug）:  clothing seg 服务图片缺少3通道逻辑增加
docs（文档变更）:
refactor（重构）:
test(增加测试):
This commit is contained in:
zchengrong
2025-06-23 15:53:41 +08:00
parent d64ffa2626
commit 87daf3b52d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/service/search_image_with_text/service.py
View File

@@ -6,7 +6,7 @@ from chromadb.config import Settings
from chromadb.utils.embedding_functions.ollama_embedding_function import OllamaEmbeddingFunction
from tqdm import tqdm
from app.core.config import OLLAMA_URL
from app.core.config import OLLAMA_URL, CHROMADB_PATH
# 读取 csv 文件
# csv_file_path = r'D:/Files/csv/output/output.csv'
@@ -15,7 +15,7 @@ from app.core.config import OLLAMA_URL
# df = pd.read_csv(csv_file_path, encoding='Windows-1252')
# 创建 Chroma 客户端
client = chromadb.Client(Settings(is_persistent=True, persist_directory="/vector_db"))
client = chromadb.Client(Settings(is_persistent=True, persist_directory=CHROMADB_PATH))
# client = chromadb.Client(Settings(is_persistent=True, persist_directory="./service/search_image_with_text/vector_db"))
# client = chromadb.Client(Settings(is_persistent=True, persist_directory="D:/workspace/AiDLab/vector_db"))
# 创建集合