update

Dockerfile

@@ -9,7 +9,6 @@ COPY ./sticker.db /tmp/sticker.db
 
 ENV HF_HOME=/tmp/.cache
 
-
 CMD python main.py
 
 EXPOSE 7860

app/config.py

@@ -5,7 +5,6 @@ DATASET_ID = "Nekoko/StickerSet"
 COZE_API_TOKEN = os.getenv('COZE_API_TOKEN')
 HUGGING_FACE_TOKEN = os.getenv('HUGGING_FACE_TOKEN')
 DEEPSEEK_API_KEY = os.getenv('DEEPSEEK_API_KEY')
-MILVUS_DB_FILE = os.getenv('MILVUS_DB_FILE', "./sticker.db")
 MILVUS_DB_URL = os.getenv('MILVUS_DB_URL', "./sticker.db")
 MILVUS_DB_TOKEN = os.getenv('MILVUS_DB_TOKEN', "")
 EMBEDDING_MODEL = 'shibing624/text2vec-base-chinese'

embedding_test.py

@@ -7,6 +7,8 @@ from pymilvus import MilvusClient, DataType
 import time
 import gradio as gr
 
+from app.config import MILVUS_DB_URL
+
 # 配置日志
 logging.basicConfig(
     level=logging.INFO,
@@ -66,7 +68,7 @@ class SentenceTransformerModel(BaseEmbeddingModel):
 class StickerSearcher:
     def __init__(self, model: BaseEmbeddingModel):
         self.model = model
-        self.client = MilvusClient(uri='./sticker.db')
+        self.client = MilvusClient(uri=MILVUS_DB_URL)
         self.collection_name = f'test_{model.model_name.replace("/", "_").replace("-", "_")}'
 
     def init_collection(self) -> bool:
@@ -158,7 +160,7 @@ def create_gradio_ui():
 
     def init_collections():
         try:
-            client = MilvusClient(uri='./sticker.db')
+            client = MilvusClient(uri=MILVUS_DB_URL)
             stickers = client.query(
                 collection_name='stickers',
                 filter='',