Spaces:
Sleeping
Sleeping
januarevan
commited on
Commit
•
12b81f7
1
Parent(s):
3e64b70
init
Browse files
main.py
CHANGED
@@ -21,15 +21,13 @@ embedding_model = SentenceTransformer('Alibaba-NLP/gte-large-en-v1.5',
|
|
21 |
)
|
22 |
collection_name="rag"
|
23 |
|
24 |
-
|
25 |
-
|
26 |
-
|
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(
|