feat design相关模型迁移4090测试

fix

app/core/config.py

@@ -135,7 +135,8 @@ SEGMENTATION = {
     "input": "seg_input__0",
     "output": "seg_output__0",
 }
-
+# ollama config
+OLLAMA_URL = "http://10.1.1.243:11434/api/embeddings"
 # DESIGN config
 DESIGN_MODEL_URL = '10.1.1.243:10000'
 AIDA_CLOTHING = "aida-clothing"

app/service/search_image_with_text/service.py

@@ -6,6 +6,8 @@ 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
+
 # 读取 csv 文件
 # csv_file_path = r'D:/Files/csv/output/output.csv'
 # image_path = r'D:/images-clean'
@@ -18,7 +20,7 @@ client = chromadb.Client(Settings(is_persistent=True, persist_directory="/vector
 # client = chromadb.Client(Settings(is_persistent=True, persist_directory="D:/workspace/AiDLab/vector_db"))
 # 创建集合
 # embedding_fn = OllamaEmbeddingFunction(url="http://localhost:11434/api/embeddings", model_name="mxbai-embed-large")
-embedding_fn = OllamaEmbeddingFunction(url="http://10.1.1.243:11434/api/embeddings", model_name="mxbai-embed-large")
+embedding_fn = OllamaEmbeddingFunction(url=OLLAMA_URL, model_name="mxbai-embed-large")
 
 
 # def create_collection():