thejagstudio commited on
Commit
0bbed5a
·
verified ·
1 Parent(s): 1a8a4da

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +8 -6
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
- persist_directory=CHROMA_PATH, embedding_function=custom_embeddings
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):