Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ def invoke(openai_api_key, youtube_url, process_video, prompt):
|
|
37 |
splits = text_splitter.split_documents(docs)
|
38 |
else:
|
39 |
print(222)
|
40 |
-
splits =
|
41 |
vector_db = Chroma.from_documents(documents = splits, embedding = OpenAIEmbeddings(), persist_directory = CHROMA_DIR)
|
42 |
llm = ChatOpenAI(model_name = MODEL_NAME, temperature = 0)
|
43 |
qa_chain = RetrievalQA.from_chain_type(llm, retriever = vector_db.as_retriever(), return_source_documents = True, chain_type_kwargs = {"prompt": QA_CHAIN_PROMPT})
|
|
|
37 |
splits = text_splitter.split_documents(docs)
|
38 |
else:
|
39 |
print(222)
|
40 |
+
splits = iter([])
|
41 |
vector_db = Chroma.from_documents(documents = splits, embedding = OpenAIEmbeddings(), persist_directory = CHROMA_DIR)
|
42 |
llm = ChatOpenAI(model_name = MODEL_NAME, temperature = 0)
|
43 |
qa_chain = RetrievalQA.from_chain_type(llm, retriever = vector_db.as_retriever(), return_source_documents = True, chain_type_kwargs = {"prompt": QA_CHAIN_PROMPT})
|