Omid-sar commited on
Commit
111a9a3
·
1 Parent(s): 574fc32

Modified Chroma variables

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -11,9 +11,7 @@ import textwrap
11
  import streamlit as st
12
 
13
 
14
- # Load OpenAI API key and define directory to store the data
15
 
16
- persist_directory = "../../data/processed"
17
 
18
  # Load the OpenAI Embeddings, LLM , PromptTemplate and LLMChain
19
  embeddings = OpenAIEmbeddings()
@@ -59,7 +57,7 @@ if video_url and question:
59
  # docs[0].page_content
60
  # Create the vector database which will be used to search for similar sentences
61
  vectordb = Chroma.from_documents(
62
- documents=docs, embedding=embeddings, persist_directory=persist_directory
63
  )
64
 
65
  # Search for the most similar sentences to the question and concatenate top 3 vectors
 
11
  import streamlit as st
12
 
13
 
 
14
 
 
15
 
16
  # Load the OpenAI Embeddings, LLM , PromptTemplate and LLMChain
17
  embeddings = OpenAIEmbeddings()
 
57
  # docs[0].page_content
58
  # Create the vector database which will be used to search for similar sentences
59
  vectordb = Chroma.from_documents(
60
+ documents=docs, embedding=embeddings, persist_directory="./chroma_db"
61
  )
62
 
63
  # Search for the most similar sentences to the question and concatenate top 3 vectors