Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -73,12 +73,14 @@ class MyEmbeddingFunction(EmbeddingFunction):
|
|
73 |
return response.json()["embeddings"][0]
|
74 |
except:
|
75 |
pass
|
76 |
-
|
77 |
-
CHROMA_PATH = "chroma"
|
78 |
-
custom_embeddings = MyEmbeddingFunction()
|
79 |
-
db = Chroma(
|
80 |
-
|
81 |
-
)
|
|
|
|
|
82 |
|
83 |
|
84 |
def embeddingGen(query):
|
|
|
73 |
return response.json()["embeddings"][0]
|
74 |
except:
|
75 |
pass
|
76 |
+
try:
|
77 |
+
CHROMA_PATH = "chroma"
|
78 |
+
custom_embeddings = MyEmbeddingFunction()
|
79 |
+
db = Chroma(
|
80 |
+
persist_directory=CHROMA_PATH, embedding_function=custom_embeddings
|
81 |
+
)
|
82 |
+
except:
|
83 |
+
pass
|
84 |
|
85 |
|
86 |
def embeddingGen(query):
|