Modified Chroma variables
Browse files
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=
|
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
|