nightfury commited on
Commit
7f3a89f
·
verified ·
1 Parent(s): f5a896d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -63,8 +63,10 @@ def query_chromadb():
63
  embeddings = OpenAIEmbeddings()
64
  # Load Vector store from local disk
65
  vectorstore = Chroma(persist_directory=DB_DIR, embedding_function=embeddings)
 
66
 
67
- result = vectorstore.similarity_search_with_score(query="who is FREDERICK?", k=4)
 
68
  jsonable_result = jsonable_encoder(result)
69
  print(json.dumps(jsonable_result, indent=2))
70
 
 
63
  embeddings = OpenAIEmbeddings()
64
  # Load Vector store from local disk
65
  vectorstore = Chroma(persist_directory=DB_DIR, embedding_function=embeddings)
66
+ db2.persist()
67
 
68
+ result = vectorstore.similarity_search_with_score(query="how to use AI in water conservation?", k=4)
69
+
70
  jsonable_result = jsonable_encoder(result)
71
  print(json.dumps(jsonable_result, indent=2))
72