januarevan commited on
Commit
12b81f7
1 Parent(s): 3e64b70
Files changed (1) hide show
  1. main.py +3 -5
main.py CHANGED
@@ -21,15 +21,13 @@ embedding_model = SentenceTransformer('Alibaba-NLP/gte-large-en-v1.5',
21
  )
22
  collection_name="rag"
23
 
24
- def setup_milvus():
25
- global milvus_client
26
- milvus_client = MilvusClientSingleton.get_instance(uri="/app/milvus_data/milvus_demo.db")
27
 
28
  def document_to_embeddings(content:str) -> list:
29
  return embedding_model.encode(content, show_progress_bar=True)
30
 
31
- setup_milvus()
32
-
33
  app = FastAPI()
34
 
35
  app.add_middleware(
 
21
  )
22
  collection_name="rag"
23
 
24
+
25
+ global milvus_client
26
+ milvus_client = MilvusClientSingleton.get_instance(uri="/app/milvus_data/milvus_demo.db")
27
 
28
  def document_to_embeddings(content:str) -> list:
29
  return embedding_model.encode(content, show_progress_bar=True)
30
 
 
 
31
  app = FastAPI()
32
 
33
  app.add_middleware(